Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

PyTorch

PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab.

https://github.com/Bil369/MaskDetect-YOLOv4-PyTorch

基于PyTorch&YOLOv4实现的口罩佩戴检测 :star: 自建口罩数据集分享

maskdetect pytorch yolov4

Last synced: 09 Nov 2024

https://github.com/Siyu-C/ACAR-Net

[CVPR 2021] Actor-Context-Actor Relation Network for Spatio-temporal Action Localization

acar-net action-detection activitynet activitynet-challenge arxiv ava ava-dataset ava-kinetics model-zoo pytorch spatio-temporal-action-localization torch

Last synced: 03 Nov 2024

https://github.com/bytedance/flux

A fast communication-overlapping library for tensor parallelism on GPUs.

cuda cutlass gpu pytorch

Last synced: 15 Nov 2024

https://github.com/islanna/dynamicrelu

Implementation of Dynamic ReLU on Pytorch

dynamic pytorch relu

Last synced: 15 Nov 2024

https://github.com/jacobgil/pytorch-zssr

PyTorch implementation of 1712.06087 "Zero-Shot" Super-Resolution using Deep Internal Learning

computer-vision deep-learning pytorch super-resolution

Last synced: 31 Oct 2024

https://ku-cvlab.github.io/Self-Attention-Guidance/

Official implementation of the paper "Improving Sample Quality of Diffusion Models Using Self-Attention Guidance" (ICCV 2023)

artificial-intelligence diffusion-models generative-models pytorch

Last synced: 31 Oct 2024

https://github.com/ssube/onnx-web

web UI for GPU-accelerated ONNX pipelines like Stable Diffusion, even on Windows and AMD

ai-art amd diffusion esrgan flask generative-art gfpgan image-generation linux nvidia python pytorch reactjs stable-diffusion super-resolution text2image upscaling web-app windows

Last synced: 14 Oct 2024

https://github.com/matthew-z/r-net

R-net in PyTorch, with ELMo

allennlp pytorch r-net squad

Last synced: 14 Oct 2024

https://github.com/hangg7/deformable-kernels

Deforming kernels to adapt towards object deformation. In ICLR 2020.

iclr2020 operator pytorch representation-learning

Last synced: 16 Nov 2024

https://github.com/yanndubs/neural-process-family

Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.

conditional-neural-process deep-learning machine-learning meta-learning neural-processes pytorch stochastic-processes uncertainty-estimation

Last synced: 15 Nov 2024

https://github.com/matthew-z/R-net

R-net in PyTorch, with ELMo

allennlp pytorch r-net squad

Last synced: 02 Nov 2024

https://github.com/sml2h3/captcha_trainer_pytorch

基于MobileNetV2/EfficientNet-b0/... + LSTM + CTC的不定长图像识别训练pytorch框架

classfication efficientnet mobilenetv2 pytorch

Last synced: 27 Oct 2024

https://github.com/kyegomez/visionmamba

Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model" It's 2.8x faster than DeiT and saves 86.8% GPU memory when performing batch inference to extract features on high-res images

ai machine-learning mamba pytorch recurrent-neural-network ssm

Last synced: 16 Nov 2024

https://github.com/sea-snell/implicit-language-q-learning

Official code from the paper "Offline RL for Natural Language Generation with Implicit Language Q Learning"

implicit-q-learning iql language-model nlp offline-rl python pytorch q-learning reinforcement-learning

Last synced: 18 Nov 2024

https://github.com/shreyas-bk/u2netdemo

Demonstration using Google Colab to show how U-2-NET can be used for Background Removal, Changing Backgrounds, Bounding Box Creation, Salient Feature Highlighting and Salient Object Cropping.

background-removal bounding-boxes deep-learning image-cropping machine-learning python pytorch saliency-map tensorflow u2net

Last synced: 25 Oct 2024

https://github.com/aws/sagemaker-pytorch-training-toolkit

Toolkit for running PyTorch training scripts on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at https://github.com/aws/deep-learning-containers.

aws docker pytorch sagemaker

Last synced: 07 Oct 2024

https://github.com/vt-vl-lab/guided-pix2pix

[ICCV 2019] Guided Image-to-Image Translation with Bi-Directional Feature Transformation

deep-learning gan pix2pix pytorch

Last synced: 15 Nov 2024

https://github.com/opendilab/di-hpc

OpenDILab RL HPC OP Lib

cuda hpc lstm pytorch reinforcement-learning

Last synced: 09 Nov 2024

https://github.com/google/jetstream

JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).

gemma gpt gpu inference jax large-language-models llama llama2 llm llm-inference llmops mlops model-serving pytorch tpu transformer

Last synced: 29 Sep 2024

https://github.com/Karel911/TRACER

TRACER: Extreme Attention Guided Salient Object Tracing Network (AAAI 2022) implementation in PyTorch

aaai-2022 aaai2022 attention attention-mechanism background-removal image-segmentation pytorch pytorch-implementation salient-object-detection

Last synced: 02 Nov 2024

https://github.com/minyoungg/pix2latent

Code for: Transforming and Projecting Images into Class-conditional Generative Networks

biggan cma-es gan inverting-images nevergrad pytorch stylegan2

Last synced: 04 Nov 2024

https://github.com/p0p4k/pflowtts_pytorch

Unofficial implementation of NVIDIA P-Flow TTS paper

deep-learning flow-matching pflow pytorch text-to-speech tts

Last synced: 08 Aug 2024

https://minyoungg.github.io/pix2latent/

Code for: Transforming and Projecting Images into Class-conditional Generative Networks

biggan cma-es gan inverting-images nevergrad pytorch stylegan2

Last synced: 11 Nov 2024

https://github.com/LeapLabTHU/EfficientTrain

1.5−3.0× lossless training or pre-training speedup. An off-the-shelf, easy-to-implement algorithm for the efficient training of foundation visual backbones.

computer-vision deep-learning efficient-training machine-learning pytorch

Last synced: 05 Nov 2024

https://github.com/dccuchile/CC6204

Material del curso de Deep Learning de la Universidad de Chile

deep-learning pytorch uchile

Last synced: 27 Oct 2024

https://github.com/a-r-j/proteinworkshop

Benchmarking framework for protein representation learning. Includes a large number of pre-training and downstream task datasets, models and training/task utilities. (ICLR 2024)

benchmark dataset deep-learning lightning pretraining protein protein-structure pytorch

Last synced: 16 Nov 2024

https://github.com/coqui-ai/trainer

🐸 - A general purpose model trainer, as flexible as it gets

ai data-science deep-learning machine-learning pytorch

Last synced: 15 Nov 2024

https://github.com/hityzy1122/opencv_transforms_torchvision

opencv reimplement for transforms in torchvision

opencv pillow pytorch torchvision

Last synced: 12 Nov 2024

https://github.com/rlaphoenix/vsgan

PyTorch-based Super-Resolution and Restoration Image Processing Module for VapourSynth

deep-learning editing esrgan gan image-manipulation python pytorch restoration upscaling vapoursynth

Last synced: 17 Nov 2024

https://github.com/eladrich/pix2vertex.pytorch

An official pyTorch port of the pix2vertex paper from ICCV2017

deep-learning iccv pytorch reconstruction

Last synced: 15 Nov 2024

https://github.com/Bobholamovic/CDLab

Yet another repository for developing and benchmarking deep learning-based change detection methods.

change-detection deep-learning pytorch remote-sensing

Last synced: 17 Nov 2024

https://github.com/bobholamovic/cdlab

Yet another repository for developing and benchmarking deep learning-based change detection methods.

change-detection deep-learning pytorch remote-sensing

Last synced: 17 Nov 2024

https://github.com/francescosaveriozuppichini/resnet

Clean, scalable and easy to use ResNet implementation in Pytorch

convolutional-neural-networks deep-learning neural-networks python3 pytorch

Last synced: 15 Nov 2024

https://github.com/jameschapman19/cca_zoo

Canonical Correlation Analysis Zoo: A collection of Regularized, Deep Learning based, Kernel, and Probabilistic methods in a scikit-learn style framework

canonical-correlation-analysis cca cca-zoo dcca deep kernel multiset-cca multiview pls pytorch tensor-cca

Last synced: 14 Nov 2024

https://github.com/chr5tphr/zennit

Zennit is a high-level framework in Python using PyTorch for explaining/exploring neural networks using attribution methods like LRP.

attribution deep-learning explainability explainable-ai feature-attribution interpretability interpretable-ai interpretable-ml lrp machine-learning python pytorch xai

Last synced: 30 Oct 2024

https://github.com/openclimatefix/graph_weather

PyTorch implementation of Ryan Keisler's 2022 "Forecasting Global Weather with Graph Neural Networks" paper (https://arxiv.org/abs/2202.07575)

forecasting-models graph-neural-networks pytorch weather

Last synced: 14 Nov 2024

https://github.com/taited/clip-score

Quick scripts to calculate CLIP text-image similarity

batch clip clip-score pytorch

Last synced: 13 Nov 2024

https://github.com/recursionpharma/gflownet

GFlowNet library specialized for graph & molecular data

deep-learning gflownet graph-neural-network pytorch

Last synced: 16 Nov 2024

https://github.com/wmcnally/golfdb

GolfDB is a video database for Golf Swing Sequencing, which involves detecting 8 golf swing events in trimmed golf swing videos. This repo demos the baseline model, SwingNet.

action-recognition event-detection golf pytorch temporal video

Last synced: 13 Aug 2024

https://github.com/pythonlessons/mltu

Machine Learning Training Utilities (for TensorFlow and PyTorch)

machine-learning object-detection ocr pytorch speech-recognition speech-to-text tensorflow yolov8

Last synced: 13 Nov 2024

https://github.com/lettucecfd/lettuce

Computational Fluid Dynamics based on PyTorch and the Lattice Boltzmann Method

cfd lattice-boltzmann machine-learning physics-simulation pytorch torch

Last synced: 05 Aug 2024

https://github.com/greydanus/mnist1d

A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.

convnet dataset machine-learning neural-networks pytorch research

Last synced: 01 Oct 2024

https://github.com/isl-org/Open3D-PointNet

Open3D PointNet implementation with PyTorch

open3d pointnet pytorch

Last synced: 28 Oct 2024

https://github.com/zjunlp/KnowPrompt

[WWW 2022] KnowPrompt: Knowledge-aware Prompt-tuning with Synergistic Optimization for Relation Extraction

dataset dialogre few-shot-learning knowledge-informed-prompt-learning knowprompt prompt prompt-learning prompt-tuning pytorch re relation-extraction semeval tacred text-classification

Last synced: 16 Nov 2024

https://github.com/locuslab/e2e-model-learning

Task-based end-to-end model learning in stochastic optimization

deep-learning machine-learning optimization paper pytorch stochastic-optimizers

Last synced: 10 Nov 2024

https://github.com/yanndubs/hash-embeddings

PyTorch implementation of Hash Embeddings (NIPS 2017). Submission to the NIPS Implementation Challenge.

embeddings hashing nips nips-challenge nlp pytorch reproducible-research word-embeddings

Last synced: 27 Oct 2024

https://github.com/LambdaLabsML/distributed-training-guide

Best practices & guides on how to write distributed pytorch training code

cluster cuda deepspeed distributed-training fsdp gpu gpu-cluster kuberentes lambdalabs mpi nccl pytorch sharding slurm

Last synced: 21 Oct 2024

https://github.com/rlaphoenix/VSGAN

PyTorch-based Super-Resolution and Restoration Image Processing Module for VapourSynth

deep-learning editing esrgan gan image-manipulation python pytorch restoration upscaling vapoursynth

Last synced: 03 Nov 2024

https://github.com/ghimiredhikura/Complex-YOLOv3

PyTorch implementation of Complex-YOLO paper with YoloV3

deep-learning object-detection pytorch

Last synced: 09 Nov 2024

https://github.com/Randl/ShuffleNetV2-pytorch

Implementation of ShuffleNetV2 for pytorch

cnn-model pytorch shufflenetv2

Last synced: 31 Oct 2024

https://github.com/yangyanli/do-conv

Depthwise Over-parameterized Convolutional Layer

convolutional-neural-networks deep-learning gluoncv pytorch tensorflow

Last synced: 18 Nov 2024

https://github.com/OlgaChernytska/word2vec-pytorch

Implementation of the first paper on word2vec

deep-learning natural-language-processing pytorch word2vec

Last synced: 26 Sep 2024

https://github.com/guotong1988/NL2SQL-RULE

Content Enhanced BERT-based Text-to-SQL Generation https://arxiv.org/abs/1910.07179

bert deep-learning knowledge knowledge-representation nl2sql nlp pytorch rule-inject-to-model semantic-parsing text2sql

Last synced: 11 Nov 2024

https://github.com/clovaai/fewshot-font-generation

The unified repository for few-shot font generation methods. This repository includes FUNIT (ICCV'19), DM-Font (ECCV'20), LF-Font (AAAI'21) and MX-Font (ICCV'21).

deep-learning fewshot-font-generation font font-generation generative-models machine-learning pytorch

Last synced: 12 Nov 2024

https://github.com/rasbt/dora-from-scratch

LoRA and DoRA from Scratch Implementations

finetuning llm pytorch

Last synced: 15 Nov 2024

https://github.com/eric-canas/qreader

Robust and Straight-Forward solution for reading difficult and tricky QR codes within images in Python. Powered by YOLOv8

computer-vision easy-to-use image-processing object-detection pip python pytorch pyzbar qr qrcode qrcode-reader qrcode-scanner yolov8

Last synced: 13 Nov 2024

https://github.com/lightforever/mlcomp

Distributed DAG (Directed acyclic graph) framework for machine learning with UI

artificial-intelligence automl computer-vision deep-learning distributed-computing infrastructure machine-learning python pytorch research

Last synced: 15 Nov 2024

https://github.com/flyingpot/pytorch_deephash

Pytorch implementation of Deep Learning of Binary Hash Codes for Fast Image Retrieval, CVPRW 2015

deep-learning deephash hash image-retrieval pytorch

Last synced: 16 Nov 2024

https://github.com/cmsflash/beauty-net

A simple, flexible, and extensible template for PyTorch. It's beautiful.

cnn computer-vision deep-learning image-classification pytorch pytorch-template

Last synced: 10 Nov 2024

https://github.com/dair-ai/emotion_dataset

:smile: Dataset for Emotion Recognition Research

dataset machine-learning nlp pytorch

Last synced: 17 Nov 2024

https://github.com/ikergarcia1996/easy-translate

Easy-Translate is a script for translating large text files with a SINGLE COMMAND. Easy-Translate is designed to be as easy as possible for beginners and as seamlesscustomizable and as possible for advanced users.

4-bit 8-bit begginers cpu easy easy-to-use gpu hugginface hugginface-hub huggingface-transformers llm m2m100 machine-translation nllb200 prompt pytorch quantization transformers translation

Last synced: 17 Nov 2024

https://github.com/csukuangfj/kaldifeat

Kaldi-compatible online & offline feature extraction with PyTorch, supporting CUDA, batch processing, chunk processing, and autograd - Provide C++ & Python API

cpp fbank features-extraction kaldi mfcc online-feature-extractor plp python pytorch streaming-feature-extractor

Last synced: 22 Oct 2024

https://github.com/li-plus/dsnet

DSNet: A Flexible Detect-to-Summarize Network for Video Summarization

computer-vision detection machine-learning pytorch video-summarization

Last synced: 06 Nov 2024

https://github.com/project-agml/agml

AgML is a centralized framework for agricultural machine learning. AgML provides access to public agricultural datasets for common agricultural deep learning tasks, with standard benchmarks and pretrained models, as well the ability to generate synthetic data and annotations.

agriculture computer-vision dataset deep-learning image-classification object-detection pytorch semantic-segmentation synthetic-data

Last synced: 11 Nov 2024

https://github.com/Project-AgML/AgML

AgML is a centralized framework for agricultural machine learning. AgML provides access to public agricultural datasets for common agricultural deep learning tasks, with standard benchmarks and pretrained models, as well the ability to generate synthetic data and annotations.

agriculture computer-vision dataset deep-learning image-classification object-detection pytorch semantic-segmentation synthetic-data

Last synced: 14 Nov 2024

https://github.com/lonepatient/lookahead_pytorch

pytorch implement of Lookahead Optimizer

cifar10 lookahead optimizer pytorch resnet-18

Last synced: 06 Nov 2024

https://github.com/fanglanting/skip-gram-pytorch

A complete pytorch implementation of skip-gram

embed pytorch spearman word2vec

Last synced: 07 Aug 2024

https://github.com/iamyuanchung/Autoregressive-Predictive-Coding

Autoregressive Predictive Coding: An unsupervised autoregressive model for speech representation learning

pre-training pytorch representation-learning self-supervised-learning unsupervised-learning

Last synced: 07 Aug 2024

https://github.com/SurajDonthi/Multi-Camera-Person-Re-Identification

State-of-the-art model for person re-identification in Multi-camera Multi-Target Tracking. Benchmarked on Market-1501 and DukeMTMTC-reID datasets.

computer-vision convolutional-neural-networks deep-learning dukemtmc-reid market-1501 mtmct multi-camera-tracking person-reidentification pytorch pytorch-lightning spatial-temporal

Last synced: 28 Oct 2024

https://github.com/tugstugi/pytorch-speech-commands

Speech commands recognition with PyTorch | Kaggle 10th place solution in TensorFlow Speech Recognition Challenge

cifar10 classification deep-learning densenet dual-path-networks kaggle neural-network pytorch resnet resnext speech-recognition wide-residual-networks

Last synced: 15 Nov 2024

https://github.com/AlanLi1997/slim-neck-by-gsconv

Pytorch implementation of the 'Slim-neck by GSConv: a lightweight-design for real-time detector architectures'

gsconv lightweight-framework objectdetection pytorch yolo yolov4 yolov5

Last synced: 09 Nov 2024

https://github.com/Audio-WestlakeU/NBSS

The official repo of NBC & SpatialNet for multichannel speech separation, denoising, and dereverberation

denoising dereverberation enhancement full-band multi-channel narrow-band pytorch separation speech

Last synced: 28 Oct 2024