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/zwwwayne/k-net

[NeurIPS2021] Code Release of K-Net: Towards Unified Image Segmentation

image-segmentation mmdetection mmsegmentation panoptic-segmentation pytorch semantic-segmentation

Last synced: 13 Jan 2025

https://github.com/hkchengrex/mivos

[CVPR 2021] Modular Interactive Video Object Segmentation: Interaction-to-Mask, Propagation and Difference-Aware Fusion. Semi-supervised VOS as well!

computer-vision cvpr2021 deep-learning interactive-segmentation pytorch segmentation video-object-segmentation video-segmentation

Last synced: 12 Jan 2025

https://github.com/Neutone/neutone_sdk

Join the community on Discord for more discussions around Neutone! https://discord.gg/VHSMzb8Wqp

ai audio audio-plugin deep-learning machine-learning music python pytorch realtime-audio sdk vst

Last synced: 05 Nov 2024

https://github.com/kevinzakka/recurrent-visual-attention

A PyTorch Implementation of "Recurrent Models of Visual Attention"

attention pytorch ram recurrent-attention-model recurrent-models

Last synced: 30 Oct 2024

https://github.com/ZwwWayne/K-Net

[NeurIPS2021] Code Release of K-Net: Towards Unified Image Segmentation

image-segmentation mmdetection mmsegmentation panoptic-segmentation pytorch semantic-segmentation

Last synced: 15 Nov 2024

https://github.com/salesforce/warp-drive

Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)

cuda deep-learning gpu high-throughput multiagent-reinforcement-learning numba pytorch reinforcement-learning

Last synced: 10 Jan 2025

https://github.com/backtime92/CRAFT-Reimplementation

CRAFT-Pyotorch:Character Region Awareness for Text Detection Reimplementation for Pytorch

craft pytorch text-detection

Last synced: 03 Nov 2024

https://github.com/bshall/knn-vc

Voice Conversion With Just Nearest Neighbors

any-to-any knn pytorch self-supervised-learning speech speech-synthesis voice-conversion

Last synced: 12 Jan 2025

https://github.com/JingyibySUTsoftware/Yolov5-deepsort-driverDistracted-driving-behavior-detection

基于深度学习的驾驶员分心驾驶行为(疲劳+危险行为)预警系统使用YOLOv5+Deepsort实现驾驶员的危险驾驶行为的预警监测

deeplearning deepsort detection dlib dlib-face-detection dlib-tracker driver-behavior object-detection opencv playsound pyqt5 python pytorch sort tired tracker yolov5 yolov5-deepsort

Last synced: 09 Nov 2024

https://github.com/criteo-research/reco-gym

Code for reco-gym: A Reinforcement Learning Environment for the problem of Product Recommendation in Online Advertising

criteo openai-gym pytorch recommendation-system

Last synced: 17 Nov 2024

https://github.com/wenmuzhou/psenet.pytorch

A pytorch re-implementation of PSENet: Shape Robust Text Detection with Progressive Scale Expansion Network

ocr python3 pytorch text-detection

Last synced: 12 Jan 2025

https://github.com/apple/ml-mdm

Train high-quality text-to-image diffusion models in a data & compute efficient manner

deep-learning diffusion-models large-scale-vision-models machine-learning pytorch

Last synced: 12 Jan 2025

https://github.com/chingyaoc/ggnn.pytorch

A PyTorch Implementation of Gated Graph Sequence Neural Networks (GGNN)

ggnn gnns graph pytorch

Last synced: 30 Oct 2024

https://github.com/edisonleeeee/graphgallery

GraphGallery is a gallery for benchmarking Graph Neural Networks, From InplusLab.

adversarial-machine-learning dgl geometric-deep-learning graph graph-neural-networks graphgallery models pyg pytorch

Last synced: 13 Jan 2025

https://github.com/vita-group/autogan

[ICCV 2019] "AutoGAN: Neural Architecture Search for Generative Adversarial Networks" by Xinyu Gong, Shiyu Chang, Yifan Jiang and Zhangyang Wang

autogan cifar10 gan neural-architecture-search pytorch

Last synced: 12 Jan 2025

https://github.com/mit-han-lab/mcunet

[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning

deep-learning microncontroller neural-architecture-search pytorch tinyml

Last synced: 28 Oct 2024

https://github.com/TRI-ML/dd3d

Official PyTorch implementation of DD3D: Is Pseudo-Lidar needed for Monocular 3D Object detection? (ICCV 2021), Dennis Park*, Rares Ambrus*, Vitor Guizilini, Jie Li, and Adrien Gaidon.

computer-vision deep-learning pytorch

Last synced: 28 Oct 2024

https://github.com/likyoo/open-cd

A Change Detection Repo Standing on the Shoulders of Giants

bit change-detection changer deep-learning pytorch transformer

Last synced: 17 Nov 2024

https://github.com/sinAshish/Multi-Scale-Attention

Code for our paper "Multi-scale Guided Attention for Medical Image Segmentation"

attention-mechanism biomedical-image-analysis deep-learning encoder-decoder pytorch

Last synced: 15 Nov 2024

https://github.com/Schlumberger/joint-vae

Pytorch implementation of JointVAE, a framework for disentangling continuous and discrete factors of variation :star2:

disentangled-representations disentanglement generative-models gumbel-softmax pytorch vae

Last synced: 14 Nov 2024

https://github.com/layumi/aicity-reid-2020

:red_car: The 1st Place Submission to AICity Challenge 2020 re-id track (Baidu-UTS submission)

aicity cityflow cvpr2020 paddlepaddle pytorch vehicle vehicle-reid veri-776

Last synced: 11 Jan 2025

https://github.com/EdisonLeeeee/GraphGallery

GraphGallery is a gallery for benchmarking Graph Neural Networks, From InplusLab.

adversarial-machine-learning dgl geometric-deep-learning graph graph-neural-networks graphgallery models pyg pytorch

Last synced: 12 Nov 2024

https://github.com/yinchangchang/ocr_densenet

第一届西安交通大学人工智能实践大赛(2018AI实践大赛--图片文字识别)第一名;仅采用densenet识别图中文字

densenet ocr ocr-recognition python pytorch

Last synced: 03 Nov 2024

https://github.com/ploomber/sklearn-evaluation

Machine learning model evaluation made easy: plots, tables, HTML reports, experiment tracking and Jupyter notebook analysis.

data-science deep-learning jupyter-notebook machine-learning pytorch scikit-learn sklearn tensorflow

Last synced: 09 Jan 2025

https://github.com/layumi/AICIty-reID-2020

:red_car: The 1st Place Submission to AICity Challenge 2020 re-id track (Baidu-UTS submission)

aicity cityflow cvpr2020 paddlepaddle pytorch vehicle vehicle-reid veri-776

Last synced: 08 Nov 2024

https://github.com/clovaai/rexnet

Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models

deep-learning image-classification imagenet object-detection pretrained-model pytorch rexnet transfer-learning

Last synced: 13 Jan 2025

https://github.com/mir-aidj/all-in-one

All-In-One Music Structure Analyzer

allin1 beat-tracking music-structure-analysis pytorch

Last synced: 22 Nov 2024

https://github.com/dfdazac/wassdistance

Approximating Wasserstein distances with PyTorch

deep-learning machine-learning optimal-transport pytorch

Last synced: 12 Nov 2024

https://github.com/walktree/libtorch-yolov3

A Libtorch implementation of the YOLO v3 object detection algorithm

cpp libtorch pytorch yolov3

Last synced: 09 Nov 2024

https://github.com/huggingface/large_language_model_training_playbook

An open collection of implementation tips, tricks and resources for training large language models

cuda large-language-models llm nccl nlp performance python pytorch scalability troubleshooting

Last synced: 11 Nov 2024

https://github.com/kubeflow/mpi-operator

Kubernetes Operator for MPI-based applications (distributed training, HPC, etc.)

apache-mxnet distributed-computing horovod kubeflow kubernetes mpi pytorch tensorflow

Last synced: 09 Jan 2025

https://github.com/aamini/evidential-deep-learning

Learn fast, scalable, and calibrated measures of uncertainty using neural networks!

confidence deep-learning evidence neural-network neurips-2020 pytorch tensorflow uncertainty

Last synced: 11 Jan 2025

https://github.com/cuiziteng/Illumination-Adaptive-Transformer

[BMVC 2022] You Only Need 90K Parameters to Adapt Light: A Light Weight Transformer for Image Enhancement and Exposure Correction. SOTA for low light enhancement, 0.004 seconds try this for pre-processing.

bmvc exposure-correction image-enhancement image-reconstruction image-restoration low-level-vision low-light-enhance low-light-image-enhancement pytorch transformer-architecture transformer-models

Last synced: 06 Nov 2024

https://github.com/pykale/pykale

Knowledge-Aware machine LEarning (KALE): accessible machine learning from multiple sources for interdisciplinary research, part of the 🔥PyTorch ecosystem. ⭐ Star to support our work!

computer-vision data-science deep-learning domain-adaptation graph-analysis knowledge-aware-learning machine-learning medical-image-analysis meta-learning multimodal multimodal-learning python pytorch transfer-learning

Last synced: 10 Jan 2025

https://github.com/chaitjo/geometric-gnn-dojo

Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks.

e3nn equivariance equivariant-networks geometric-deep-learning geometric-graphs graph-isomorphism graph-neural-networks pytorch pytorch-geometric

Last synced: 11 Jan 2025

https://github.com/loretoparisi/CapsNet

CapsNet (Capsules Net) in Geoffrey E Hinton paper "Dynamic Routing Between Capsules" - State Of the Art

capsnet capsule-networks chainer deeplearning keras lasagne machine-learning mnist mxnet pytorch tensorflow

Last synced: 02 Nov 2024

https://github.com/RaivoKoot/Video-Dataset-Loading-Pytorch

Generic PyTorch dataset implementation to load and augment VIDEOS for deep learning training loops.

action-recognition dataloader deep-learning machine-learning pytorch video-dataset videos

Last synced: 10 Nov 2024

https://github.com/mv-lab/InstructIR

[ECCV 2024] InstructIR: High-Quality Image Restoration Following Human Instructions https://huggingface.co/spaces/marcosv/InstructIR

computer-vision deblurring deep-learning dehazing denoising image-enhancement image-restoration inverse-problems language-model low-light-image-enhancement multi-task multimodal neural-network photography prompt pytorch super-resolution

Last synced: 03 Nov 2024

https://github.com/sacmehta/ESPNetv2

A light-weight, power efficient, and general purpose convolutional neural network

cnn convolutional-neural-networks deep-learning efficient imagenet lightweight machine-learning pytorch semantic-segmentation

Last synced: 31 Oct 2024

https://github.com/nschaetti/EchoTorch

A Python toolkit for Reservoir Computing and Echo State Network experimentation based on pyTorch. EchoTorch is the only Python module available to easily create Deep Reservoir Computing models.

artificial-intelligence artificial-neural-networks echo-state-networks machine-learning machine-learning-algorithms machinelearning neural-networks python python-toolkit pytorch recurrent-networks recurrent-neural-networks reservoir-computing torch

Last synced: 16 Nov 2024

https://github.com/iBelieveCJM/Tricks-of-Semi-supervisedDeepLeanring-Pytorch

PseudoLabel 2013, VAT, PI model, Tempens, MeanTeacher, ICT, MixMatch, FixMatch

fixmatch ict meanteacher mixmatch pi pseudo-label pytorch semi-supervised-learning ssl tempens vat

Last synced: 15 Nov 2024

https://github.com/lemonhu/ner-bert-pytorch

PyTorch solution of named entity recognition task Using Google AI's pre-trained BERT model.

chinese-ner entity-extraction google-bert information-extraction msra named-entity-recognition ner pytorch transformer

Last synced: 12 Jan 2025

https://github.com/datawhalechina/torch-rechub

A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.

ctr-prediction pytorch recommendation-system recsys

Last synced: 13 Jan 2025

https://github.com/yeyupiaoling/audioclassification-pytorch

The Pytorch implementation of sound classification supports EcapaTdnn, PANNS, TDNN, Res2Net, ResNetSE and other models, as well as a variety of preprocessing methods.

audioclassification ecapa-tdnn panns pytorch res2net resnet-se urbansound8k

Last synced: 10 Jan 2025

https://github.com/ruiminshen/yolo2-pytorch

PyTorch implementation of the YOLO (You Only Look Once) v2

caffe2 computer-vision deep-learning deep-neural-networks object-detection onnx python pytorch yolo2

Last synced: 04 Nov 2024

https://github.com/cfzd/Ultra-Fast-Lane-Detection-v2

Ultra Fast Deep Lane Detection With Hybrid Anchor Driven Ordinal Classification (TPAMI 2022)

deep-learning lane-detection pytorch

Last synced: 11 Nov 2024

https://github.com/lemonhu/NER-BERT-pytorch

PyTorch solution of named entity recognition task Using Google AI's pre-trained BERT model.

chinese-ner entity-extraction google-bert information-extraction msra named-entity-recognition ner pytorch transformer

Last synced: 02 Nov 2024

https://github.com/allentdan/libtorchsegmentation

A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: VGG, ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.

cpp deeplabv3 deeplabv3plus fpn image-segmentation imagenet libtorch libtorch-segment models neural-network pretrained-backbones pretrained-weights pspnet pytorch pytorch-cpp pytorch-cpp-frontend resnet resnext semantic-segmentation unet

Last synced: 12 Jan 2025

https://github.com/daniilrobnikov/vits2

VITS2: Improving Quality and Efficiency of Single-Stage Text-to-Speech with Adversarial Learning and Architecture Design

deep-learning pytorch speech speech-synthesis text-to-speech tts vits2 voice-conversion

Last synced: 27 Oct 2024

https://github.com/tlkh/ai-lab

All-in-one AI container for rapid prototyping

cuda data-science deep-learning docker jupyter nvidia pytorch tensorflow

Last synced: 12 Jan 2025

https://github.com/krasserm/perceiver-io

A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training

deep-learning machine-learning perceiver perceiver-ar perceiver-io pytorch pytorch-lightning

Last synced: 08 Jan 2025

https://github.com/blackhc/toma

Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory

data-science gpu machine-learning python pytorch

Last synced: 11 Jan 2025

https://github.com/adefossez/julius

Fast PyTorch based DSP for audio and 1D signals

dsp pytorch

Last synced: 12 Jan 2025

https://github.com/skhadem/3D-BoundingBox

PyTorch implementation for 3D Bounding Box Estimation Using Deep Learning and Geometry

3d computer-vision localization pytorch

Last synced: 28 Oct 2024

https://github.com/cccntu/minlora

minLoRA: a minimal PyTorch library that allows you to apply LoRA to any PyTorch model.

fastai huggingface pytorch pytorch-implementation pytorch-lightning

Last synced: 09 Nov 2024

https://github.com/M-Nauta/TCDF

Temporal Causal Discovery Framework (PyTorch): discovering causal relationships between time series

causality cnn forecasting machine-learning neural-networks prediction python pytorch timeseries

Last synced: 15 Nov 2024

https://github.com/omerbsezer/fast-pytorch

Pytorch Tutorial, Pytorch with Google Colab, Pytorch Implementations: CNN, RNN, DCGAN, Transfer Learning, Chatbot, Pytorch Sample Codes

cnn cnn-pytorch colab-notebook colab-tutorial colaboratory computer-vision deep-learning machine-learning mlp python pytorch pytorch-implementation pytorch-tutorial rnn rnn-pytorch transfer-learning

Last synced: 13 Jan 2025

https://github.com/ashishpatel26/Real-time-ML-Project

A curated list of applied machine learning and data science notebooks and libraries across different industries.

application deep-learning deeplearning dl keras machine-learning machine-learning-algorithms machinelearning ml ml-application project pytorch real-time real-time-data rl tensorflow theano

Last synced: 01 Nov 2024

https://github.com/BlackHC/toma

Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory

data-science gpu machine-learning python pytorch

Last synced: 15 Nov 2024

https://github.com/swz30/mirnetv2

[TPAMI 2022] Learning Enriched Features for Fast Image Restoration and Enhancement. Results on Defocus Deblurring, Denoising, Super-resolution, and image enhancement

contrast-enhancement defocus-deblurring denoising high-resolution image-denoising image-enhancement image-processing image-restoration low-level-vision low-light-image-enhancement pytorch tpami

Last synced: 12 Jan 2025

https://github.com/zihangJiang/TokenLabeling

Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"

imagenet lv-vit pytorch segmentation transformer vision

Last synced: 13 Nov 2024

https://github.com/shm007g/LLaMA-Cult-and-More

Large Language Models for All, 🦙 Cult and More, Stay in touch !

alpaca chatgpt deepspeed ggml gpt gpt4 gptq llama llm loralib pytorch tensorflow transformers vicuna

Last synced: 27 Oct 2024

https://github.com/petuum/adaptdl

Resource-adaptive cluster scheduler for deep learning training.

aws cloud deep-learning distributed-systems distributed-training kubernetes machine-learning pytorch

Last synced: 16 Nov 2024

https://github.com/vita-group/shapematchinggan

[ICCV 2019, Oral] Controllable Artistic Text Style Transfer via Shape-Matching GAN

gans iccv19 pytorch style-transfer

Last synced: 13 Jan 2025

https://github.com/KinglittleQ/torch-batch-svd

A 100x faster SVD for PyTorch⚡️

cusolver pytorch svd

Last synced: 27 Oct 2024

https://github.com/jeff-sjtu/res-loglikelihood-regression

Code for "Human Pose Regression with Residual Log-likelihood Estimation", ICCV 2021 Oral

2d-human-pose 3d-human-pose human-pose-estimation iccv iccv2021 pytorch regression

Last synced: 25 Dec 2024

https://github.com/jeffffffli/res-loglikelihood-regression

Code for "Human Pose Regression with Residual Log-likelihood Estimation", ICCV 2021 Oral

2d-human-pose 3d-human-pose human-pose-estimation iccv iccv2021 pytorch regression

Last synced: 08 Jan 2025

https://github.com/Eromera/erfnet_pytorch

Pytorch code for semantic segmentation using ERFNet

cityscapes erfnet pytorch segmentation semantic semantic-segmentation

Last synced: 18 Nov 2024

https://github.com/continvvm/continuum

A clean and simple data loading library for Continual Learning

continual-learning dataloader dataset incremental-learning lifelong-learning online-learning pytorch

Last synced: 13 Jan 2025

https://github.com/Stanford-TML/EDGE

Official PyTorch Implementation of EDGE (CVPR 2023)

animation dance-generation diffusion-models pytorch

Last synced: 14 Nov 2024

https://github.com/patrick-kidger/torchcde

Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.

controlled-differential-equations deep-learning deep-neural-networks differential-equations dynamical-systems machine-learning neural-differential-equations neural-networks pytorch time-series

Last synced: 12 Jan 2025

https://github.com/abhiskk/fast-neural-style

pytorch implementation of fast-neural-style

deep-learning neural-style pytorch

Last synced: 06 Nov 2024

https://github.com/shinya7y/UniverseNet

USB: Universal-Scale Object Detection Benchmark (BMVC 2022)

mmdetection object-detection pytorch waymo-open-dataset

Last synced: 28 Oct 2024

https://github.com/Jeff-sjtu/res-loglikelihood-regression

Code for "Human Pose Regression with Residual Log-likelihood Estimation", ICCV 2021 Oral

2d-human-pose 3d-human-pose human-pose-estimation iccv iccv2021 pytorch regression

Last synced: 18 Nov 2024