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/610265158/Peppa_Pig_Face_Landmark

A simple face detect and alignment method, which is easy and stable.

face-alignment face-keypoint face-mask facedetect keypoint pytorch

Last synced: 13 Nov 2024

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

[ICLR 2020] "FasterSeg: Searching for Faster Real-time Semantic Segmentation" by Wuyang Chen, Xinyu Gong, Xianming Liu, Qian Zhang, Yuan Li, Zhangyang Wang

cityscapes latency neural-architecture-search pytorch semantic-segmentation tensorrt

Last synced: 05 Jan 2025

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

🔥 A collection of PyTorch notebooks for learning and practicing deep learning

deep-learning machine-learning nlp pytorch

Last synced: 27 Dec 2024

https://github.com/rowanz/neural-motifs

Code for Neural Motifs: Scene Graph Parsing with Global Context (CVPR 2018)

pytorch scene-graph vision visual-genome

Last synced: 02 Nov 2024

https://github.com/facebookresearch/neuralcompression

A collection of tools for neural compression enthusiasts.

compression deep-learning jax machine-learning neural-compression python pytorch

Last synced: 10 Jan 2025

https://github.com/bradyz/cross_view_transformers

Cross-view Transformers for real-time Map-view Semantic Segmentation (CVPR 2022 Oral)

cvpr2022 deep-learning pytorch transformer

Last synced: 28 Oct 2024

https://github.com/locuslab/optnet

OptNet: Differentiable Optimization as a Layer in Neural Networks

deep-learning machine-learning optimization paper pytorch

Last synced: 06 Jan 2025

https://github.com/znxlwm/pytorch-mnist-celeba-gan-dcgan

Pytorch implementation of Generative Adversarial Networks (GAN) and Deep Convolutional Generative Adversarial Networks (DCGAN) for MNIST and CelebA datasets

celeba dcgan gan generative-adversarial-network mnist pytorch

Last synced: 04 Jan 2025

https://github.com/allenai/allennlp-models

Officially supported AllenNLP models

allennlp nlp pytorch

Last synced: 26 Sep 2024

https://github.com/HoloClean/holoclean

A Machine Learning System for Data Enrichment.

data-enrichment data-science inference-engine machine-learning pytorch

Last synced: 12 Nov 2024

https://github.com/chrischoy/pytorch-custom-cuda-tutorial

Tutorial for building a custom CUDA function for Pytorch

python pytorch pytorch-backend tutorial wrapper

Last synced: 06 Jan 2025

https://github.com/swz30/cycleisp

[CVPR 2020--Oral] CycleISP: Real Image Restoration via Improved Data Synthesis

camera-imaging-pipeline computer-vision cvpr2020 cycleisp data-synthesis image-denoising image-restoration low-level-vision pytorch raw2rgb rgb2raw

Last synced: 04 Jan 2025

https://github.com/Oldpan/Faceswap-Deepfake-Pytorch

Faceswap with Pytorch or DeepFake with Pytorch

faceswap pytorch

Last synced: 08 Nov 2024

https://github.com/hazdzz/stgcn

The PyTorch implementation of STGCN.

gcn gnn pytorch road-traffic-prediction tcn

Last synced: 08 Jan 2025

https://github.com/DagnyT/hardnet

Hardnet descriptor model - "Working hard to know your neighbor's margins: Local descriptor learning loss"

computer-vision convolutional-neural-networks deep-learning image-matching image-retrieval metric-learning nips-2017 pytorch

Last synced: 04 Nov 2024

https://github.com/xiaoyufenfei/LEDNet

LEDNet: A Lightweight Encoder-Decoder Network for Real-time Semantic Segmentation

cityscape-dataset computer-vision lednet pytorch real-time semantic-segmentation

Last synced: 14 Nov 2024

https://github.com/prlz77/resnext.pytorch

Reproduces ResNet-V3 with pytorch

cifar pytorch resnet resnext

Last synced: 07 Jan 2025

https://github.com/fancompute/wavetorch

🌊 Numerically solving and backpropagating through the wave equation

differential-equations pytorch recurrent-neural-network rnn time-series vowel-recognition wave-equation

Last synced: 14 Nov 2024

https://github.com/cardwing/codes-for-pvkd

Point-to-Voxel Knowledge Distillation for LiDAR Semantic Segmentation (CVPR 2022)

cvpr2022 lidar-segmentation nuscenes point-cloud-segmentation pytorch range-image semantickitti waymo-dataset

Last synced: 11 Jan 2025

https://github.com/jaketae/storyteller

Multimodal AI Story Teller, built with Stable Diffusion, GPT, and neural text-to-speech

ddpm diffusion-models gpt image-generation natural-language-generation pytorch stable-diffusion text-to-image text-to-speech text-to-video video-generation

Last synced: 05 Jan 2025

https://github.com/WangQvQ/YOLOMagic

YOLO Magic🪄 is an extension based on Ultralytics' YOLOv5, designed to provide more powerful functionality and simpler operations for visual tasks.

deep-learning gradio machine-learning onnx pytorch tflite yolo yolov5

Last synced: 09 Nov 2024

https://github.com/allenai/tango

Organize your experiments into discrete steps that can be cached and reused throughout the lifetime of your research project.

ai machine-learning nlp python python3 pytorch

Last synced: 01 Oct 2024

https://github.com/facebookresearch/NeuralCompression

A collection of tools for neural compression enthusiasts.

compression deep-learning jax machine-learning neural-compression python pytorch

Last synced: 23 Nov 2024

https://github.com/prlz77/ResNeXt.pytorch

Reproduces ResNet-V3 with pytorch

cifar pytorch resnet resnext

Last synced: 31 Oct 2024

https://github.com/understandable-machine-intelligence-lab/Quantus

Quantus is an eXplainable AI toolkit for responsible evaluation of neural network explanations

deep-learning explainable-ai interpretability machine-learning pytorch quantification-evaluation-methods reproducibility tensorflow xai

Last synced: 15 Nov 2024

https://github.com/harry24k/bayesian-neural-network-pytorch

PyTorch implementation of bayesian neural network [torchbnn]

bayesian deep-learning neural-network pytorch

Last synced: 04 Jan 2025

https://github.com/medipixel/rl_algorithms

Structural implementation of RL key algorithms

deep-learning dqn gym policy-gradient python3 pytorch reinforcement-learning

Last synced: 11 Nov 2024

https://github.com/hfslyc/AdvSemiSeg

Adversarial Learning for Semi-supervised Semantic Segmentation, BMVC 2018

adversarial-learning computer-vision deep-learning pytorch semantic-segmentation semi-supervised-learning

Last synced: 28 Oct 2024

https://github.com/yeyupiaoling/Whisper-Finetune

Fine-tune the Whisper speech recognition model to support training without timestamp data, training with timestamp data, and training without speech data. Accelerate inference and support Web deployment, Windows desktop deployment, and Android deployment

android asr chinese ctranslate2 huggingface lora pytorch speech-recognition transformers web whisper

Last synced: 09 Oct 2024

https://github.com/shibing624/pytextclassifier

pytextclassifier is a toolkit for text classification. 文本分类,LR,Xgboost,TextCNN,FastText,TextRNN,BERT等分类模型实现,开箱即用。

bert classification focalloss-pytorch hierarchical machine-learning nlp pytextclassifier python pytorch softmax text-classification text-classifier

Last synced: 10 Jan 2025

https://github.com/znxlwm/pytorch-mnist-celeba-cgan-cdcgan

Pytorch implementation of conditional Generative Adversarial Networks (cGAN) and conditional Deep Convolutional Generative Adversarial Networks (cDCGAN) for MNIST dataset

cdcgan celeba cgan conditional-dcgan conditional-gan gender generative-adversarial-network mnist pytorch

Last synced: 06 Jan 2025

https://github.com/v-iashin/video_features

Extract video features from raw videos using multiple GPUs. We support RAFT flow frames as well as S3D, I3D, R(2+1)D, VGGish, CLIP, and TIMM models.

audio-features clip feature-extraction i3d ig65m laion multi-gpu optical-flow parallel pytorch r2plus1d raft resnet s3d swin timm vggish video-features visual-features vit

Last synced: 03 Nov 2024

https://github.com/shayneobrien/generative-models

Annotated, understandable, and visually interpretable PyTorch implementations of: VAE, BIRVAE, NSGAN, MMGAN, WGAN, WGANGP, LSGAN, DRAGAN, BEGAN, RaGAN, InfoGAN, fGAN, FisherGAN

autoencoder began discriminator dragan fgan fishergan gan generative-adversarial-network generative-models infogan lsgan machine-learning mmgan nsgan python pytorch ragan vae wasserstein wgan

Last synced: 02 Nov 2024

https://github.com/ldeecke/gmm-torch

Gaussian mixture models in PyTorch.

gaussian-mixture-models pytorch

Last synced: 23 Nov 2024

https://github.com/ntmc-community/matchzoo-py

Facilitating the design, comparison and sharing of deep text matching models.

deep-learning matching natural-language-processing neural-network pytorch text text-matching

Last synced: 05 Jan 2025

https://github.com/yihongXU/deepMOT

Official Implementation of How To Train Your Deep Multi-Object Tracker (CVPR2020)

deep-learning multi-object-tracking python pytorch

Last synced: 13 Nov 2024

https://github.com/cardwing/Codes-for-PVKD

Point-to-Voxel Knowledge Distillation for LiDAR Semantic Segmentation (CVPR 2022)

cvpr2022 lidar-segmentation nuscenes point-cloud-segmentation pytorch range-image semantickitti waymo-dataset

Last synced: 28 Oct 2024

https://github.com/swz30/CycleISP

[CVPR 2020--Oral] CycleISP: Real Image Restoration via Improved Data Synthesis

camera-imaging-pipeline computer-vision cvpr2020 cycleisp data-synthesis image-denoising image-restoration low-level-vision pytorch raw2rgb rgb2raw

Last synced: 03 Nov 2024

https://github.com/NTMC-Community/MatchZoo-py

Facilitating the design, comparison and sharing of deep text matching models.

deep-learning matching natural-language-processing neural-network pytorch text text-matching

Last synced: 15 Nov 2024

https://github.com/PennyLaneAI/qml

Introductions to key concepts in quantum programming, as well as tutorials and implementations from cutting-edge quantum computing research.

autograd automatic-differentiation demo key-concepts neural-networks pytorch qml quantum-chemistry quantum-computing quantum-machine-learning tensorflow tutorials

Last synced: 03 Nov 2024

https://github.com/DIYer22/boxx

Tool-box for efficient build and debug in Python. Especially for Scientific Computing and Computer Vision.

awesome-python computer-vision debug debugging deep-learning hack pytorch pytorch-debug scientific-computing toolbox

Last synced: 27 Nov 2024

https://github.com/dongminlee94/deep_rl

PyTorch implementation of deep reinforcement learning algorithms

a2c ddpg ddqn deep-reinforcement-learning dqn model-free-rl npg ppo pytorch sac sac-aea td3 trpo vpg

Last synced: 06 Jan 2025

https://github.com/ksw0306/FloWaveNet

A Pytorch implementation of "FloWaveNet: A Generative Flow for Raw Audio"

clarinet generative-flow glow pytorch wavenet

Last synced: 17 Nov 2024

https://github.com/swanhubx/swanlab

⚡️SwanLab: your ML experiment notebook. 你的AI实验笔记本,日志记录与可视化AI训练全流程。

data-science deep-learning fastapi jax machine-learning mlops model-versioning python pytorch tensorboard tensorflow tracking transformers visualization

Last synced: 09 Jan 2025

https://github.com/adambielski/CapsNet-pytorch

PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules

capsnet capsules deep-learning dynamic-routing-between-capsules machine-learning mnist pytorch

Last synced: 27 Nov 2024

https://github.com/adambielski/capsnet-pytorch

PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules

capsnet capsules deep-learning dynamic-routing-between-capsules machine-learning mnist pytorch

Last synced: 05 Jan 2025

https://github.com/fartashf/vsepp

PyTorch Code for the paper "VSE++: Improving Visual-Semantic Embeddings with Hard Negatives"

bmvc negatives paper pytorch vse

Last synced: 15 Nov 2024

https://github.com/fkodom/fft-conv-pytorch

Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.

convolution image-processing neural-networks python3 pytorch

Last synced: 05 Jan 2025

https://github.com/wvangansbeke/sparse-depth-completion

Predict dense depth maps from sparse and noisy LiDAR frames guided by RGB images. (Ranked 1st place on KITTI) [MVA 2019]

computer-vision deep-learning depth-completion depth-prediction kitti lidar noisy-data pytorch sensor-fusion

Last synced: 05 Jan 2025

https://github.com/salesforce/matchbox

Write PyTorch code at the level of individual examples, then run it efficiently on minibatches.

deep-learning minibatch nlp pytorch

Last synced: 14 Nov 2024

https://github.com/thudm/graphmae

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

deep-learning graph-neural-networks pytorch self-supervised-learning

Last synced: 05 Jan 2025

https://github.com/davidefiocco/streamlit-fastapi-model-serving

Simple web app example serving a PyTorch model using streamlit and FastAPI

docker-compose fastapi pycones pytorch streamlit

Last synced: 11 Jan 2025

https://github.com/um-arm-lab/pytorch_mppi

Model Predictive Path Integral (MPPI) with approximate dynamics implemented in pytorch

approximate-dynamics controls model-predictive-control mppi pytorch

Last synced: 04 Jan 2025

https://github.com/sthalles/pytorch-byol

PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning

byol deep-learning pytorch representation-learning self-supervised-learning simclr unsupervised-learning

Last synced: 05 Jan 2025

https://github.com/crowsonkb/style-transfer-pytorch

Neural style transfer in PyTorch.

neural-art pytorch style-transfer

Last synced: 05 Jan 2025

https://github.com/nv-tlabs/steal

STEAL - Learning Semantic Boundaries from Noisy Annotations (CVPR 2019)

annotation cvpr2019 deep-learning devil-is-in-the-edges nv-tlabs pytorch semantic-boundaries semantic-segmentation steal

Last synced: 04 Jan 2025

https://github.com/THUDM/GraphMAE

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

deep-learning graph-neural-networks pytorch self-supervised-learning

Last synced: 28 Nov 2024

https://github.com/songquanpeng/pytorch-template

To be the world's best PyTorch project template.

boilerplates gan gans pytorch pytorch-template

Last synced: 05 Jan 2025

https://github.com/cure-lab/MagicDrive

[ICLR24] Official implementation of the paper “MagicDrive: Street View Generation with Diverse 3D Geometry Control”

autonomous-vehicles deep-learning diffusion-models image-generation pytorch video-generation

Last synced: 28 Oct 2024

https://github.com/ma-xu/pointMLP-pytorch

[ICLR 2022 poster] Official PyTorch implementation of "Rethinking Network Design and Local Geometry in Point Cloud: A Simple Residual MLP Framework"

modelnet40 pointcloud pytorch scanobjectnn

Last synced: 28 Oct 2024

https://github.com/wildltr/ptranking

Learning to Rank in PyTorch

learning-to-rank pytorch ranking

Last synced: 11 Nov 2024

https://github.com/archinetai/surgeon-pytorch

A library to inspect and extract intermediate layers of PyTorch models.

artificial-intelligence deep-learning pytorch

Last synced: 11 Jan 2025

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: 05 Jan 2025

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: 06 Jan 2025

https://github.com/microsoft/archai

Accelerate your Neural Architecture Search (NAS) through fast, reproducible and modular research.

automated-machine-learning automl darts deep-learning hyperparameter-optimization machine-learning model-compression nas neural-architecture-search petridish python pytorch

Last synced: 10 Jan 2025

https://github.com/tunz/transformer-pytorch

Transformer implementation in PyTorch.

pytorch transformer

Last synced: 06 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/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: 05 Jan 2025

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