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/stonesjtu/pytorch_memlab

Profiling and inspecting memory in pytorch

cuda-memory memory-profiler pytorch

Last synced: 25 Sep 2024

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

[ICCV 2019] "DeblurGAN-v2: Deblurring (Orders-of-Magnitude) Faster and Better" by Orest Kupyn, Tetiana Martyniuk, Junru Wu, Zhangyang Wang

deblurgan deep-learning generative-adversarial-network iccv iccv2019 low-level-vision pytorch ukraine

Last synced: 24 Sep 2024

https://github.com/wpeebles/gangealing

Official PyTorch Implementation of "GAN-Supervised Dense Visual Alignment" (CVPR 2022 Oral, Best Paper Finalist)

alignment computer-vision correspondence cvpr cvpr2022 deep-learning gan generative image-manipulation pytorch stylegan stylegan2 tracking

Last synced: 24 Sep 2024

https://github.com/sniklaus/sepconv-slomo

an implementation of Video Frame Interpolation via Adaptive Separable Convolution using PyTorch

cuda cupy deep-learning python pytorch

Last synced: 24 Sep 2024

https://github.com/uvipen/Super-mario-bros-A3C-pytorch

Asynchronous Advantage Actor-Critic (A3C) algorithm for Super Mario Bros

a3c ai deep-learning gym mario python pytorch reinforcement-learning super-mario-bros

Last synced: 07 Aug 2024

https://github.com/VITA-Group/DeblurGANv2

[ICCV 2019] "DeblurGAN-v2: Deblurring (Orders-of-Magnitude) Faster and Better" by Orest Kupyn, Tetiana Martyniuk, Junru Wu, Zhangyang Wang

deblurgan deep-learning generative-adversarial-network iccv iccv2019 low-level-vision pytorch ukraine

Last synced: 07 Aug 2024

https://github.com/piddnad/DDColor

[ICCV 2023] DDColor: Towards Photo-Realistic Image Colorization via Dual Decoders

computer-vision image-colorization pytorch

Last synced: 29 Aug 2024

https://github.com/Stonesjtu/pytorch_memlab

Profiling and inspecting memory in pytorch

cuda-memory memory-profiler pytorch

Last synced: 02 Aug 2024

https://github.com/ClementPinard/SfmLearner-Pytorch

Pytorch version of SfmLearner from Tinghui Zhou et al.

depth disparity kitti pytorch unsupervised

Last synced: 31 Jul 2024

https://github.com/pbloem/former

Simple transformer implementation from scratch in pytorch.

machine-learning pytorch transformer

Last synced: 25 Sep 2024

https://github.com/sicara/easy-few-shot-learning

Ready-to-use code and tutorial notebooks to boost your way into few-shot learning for image classification.

deep-learning few-shot-classifcation few-shot-learning few-shot-recognition image-classification machine-learning meta-learning pytorch

Last synced: 25 Sep 2024

https://github.com/anibali/docker-pytorch

A Docker image for PyTorch

cuda docker docker-image pytorch

Last synced: 24 Sep 2024

https://github.com/orobix/Prototypical-Networks-for-Few-shot-Learning-PyTorch

Implementation of Prototypical Networks for Few Shot Learning (https://arxiv.org/abs/1703.05175) in Pytorch

cnn prototypical-networks python pytorch

Last synced: 30 Jul 2024

https://github.com/NATSpeech/NATSpeech

A Non-Autoregressive Text-to-Speech (NAR-TTS) framework, including official PyTorch implementation of PortaSpeech (NeurIPS 2021) and DiffSpeech (AAAI 2022)

diffsinger diffspeech huggingface portaspeech pytorch speech speech-synthesis tts

Last synced: 07 Aug 2024

https://github.com/Oldpan/Pytorch-Memory-Utils

pytorch memory track code

gpu memory pytorch

Last synced: 03 Aug 2024

https://github.com/freewym/espresso

Espresso: A Fast End-to-End Neural Speech Recognition Toolkit

asr end-to-end fairseq kaldi python pytorch speech-recognition

Last synced: 01 Aug 2024

https://github.com/ermongroup/SDEdit

PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations

controllable-generation image-editing image-generation image-manipulation pytorch score-matching

Last synced: 31 Jul 2024

https://github.com/aws/deep-learning-containers

AWS Deep Learning Containers (DLCs) are a set of Docker images for training and serving models in TensorFlow, TensorFlow 2, PyTorch, and MXNet.

aws docker mxnet pytorch sagemaker tensorflow tensorflow2

Last synced: 01 Aug 2024

https://github.com/rusty1s/pytorch_sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

autograd pytorch sparse sparse-matrices

Last synced: 02 Aug 2024

https://github.com/kmkurn/pytorch-crf

(Linear-chain) Conditional random field in PyTorch.

conditional-random-fields neural-networks pytorch

Last synced: 31 Jul 2024

https://github.com/graykode/gpt-2-Pytorch

Simple Text-Generator with OpenAI gpt-2 Pytorch Implementation

gpt-2 gpt2 implementation natural-language-processing nlp pytorch story-telling text-generator

Last synced: 30 Jul 2024

https://github.com/THUDM/CogView2

official code repo for paper "CogView2: Faster and Better Text-to-Image Generation via Hierarchical Transformers"

pretrained-models pytorch text-to-image transformer

Last synced: 01 Aug 2024

https://github.com/descriptinc/melgan-neurips

GAN-based Mel-Spectrogram Inversion Network for Text-to-Speech Synthesis

deep-learning gans pytorch speech-synthesis

Last synced: 05 Aug 2024

https://github.com/vikparuchuri/zero_to_gpt

Go from no deep learning knowledge to implementing GPT.

deep-learning python pytorch tutorial

Last synced: 02 Aug 2024

https://github.com/sukebenet/dreampower

DeepNude algorithm fork with improvements.

deepnude dreamnet dreampower fake-nudes gan image-processing opencv photo pytorch

Last synced: 01 Aug 2024

https://github.com/uber/neuropod

A uniform interface to run deep learning models from multiple frameworks

deep-learning deeplearning incubation inference keras machine-learning machinelearning pytorch tensorflow

Last synced: 02 Aug 2024

https://github.com/open-mmlab/mmflow

OpenMMLab optical flow toolbox and benchmark

openmmlab optical-flow pytorch

Last synced: 31 Jul 2024

https://github.com/piesposito/blitz-bayesian-deep-learning

A simple and extensible library to create Bayesian Neural Network layers on PyTorch.

bayesian-deep-learning bayesian-layers bayesian-neural-networks pytorch pytorch-implementation pytorch-tutorial

Last synced: 25 Sep 2024

https://github.com/D-X-Y/landmark-detection

Four landmark detection algorithms, implemented in PyTorch.

facial-landmarks keypoint-detection landmark-detection pytorch style-transfer

Last synced: 31 Jul 2024

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

Gated-Shape CNN for Semantic Segmentation (ICCV 2019)

computer-vision deep-learning iccv2019 nv-tlabs pytorch semantic-boundaries semantic-segmentation

Last synced: 31 Jul 2024

https://github.com/lhotse-speech/lhotse

Tools for handling speech data in machine learning projects.

ai audio data deep-learning kaldi machine-learning python pytorch speech speech-recognition

Last synced: 08 Aug 2024

https://github.com/Sharpiless/Yolov5-Deepsort

最新版本yolov5+deepsort目标检测和追踪,能够显示目标类别,支持5.0版本可训练自己数据集

computer-vision deepsort object-detection object-tracking pytorch yolov5

Last synced: 02 Aug 2024

https://github.com/huggingface/torchMoji

😇A pyTorch implementation of the DeepMoji model: state-of-the-art deep learning model for analyzing sentiment, emotion, sarcasm etc

deep-learning machine-learning natural-language-processing python pytorch

Last synced: 01 Aug 2024

https://github.com/Mukosame/Zooming-Slow-Mo-CVPR-2020

Fast and Accurate One-Stage Space-Time Video Super-Resolution (accepted in CVPR 2020)

cvpr cvpr2020 pytorch spatio-temporal super-resolution video video-frame-interpolation video-super-resolution

Last synced: 31 Jul 2024

https://github.com/sooftware/conformer

[Unofficial] PyTorch implementation of "Conformer: Convolution-augmented Transformer for Speech Recognition" (INTERSPEECH 2020)

asr augmented cnn conformer conv convolution pytorch recognition speech speech-recognition transformer transformer-xl

Last synced: 08 Aug 2024

https://github.com/fbcotter/pytorch_wavelets

Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet

dtcwt filterbank pytorch wavelet wavelet-transform

Last synced: 31 Jul 2024

https://github.com/atong01/conditional-flow-matching

TorchCFM: a Conditional Flow Matching library

continuous-normalizing-flows optimal-transport pytorch

Last synced: 05 Aug 2024

https://github.com/jinfagang/alfred

alfred-py: A deep learning utility library for **human**, more detail about the usage of lib to: https://zhuanlan.zhihu.com/p/341446046

3d-object-detection alfred alfred-py deeplearning network pycocotools pytorch segmentation sensor-fusion video-combiner

Last synced: 08 Aug 2024

https://github.com/lucasjinreal/alfred

alfred-py: A deep learning utility library for **human**, more detail about the usage of lib to: https://zhuanlan.zhihu.com/p/341446046

3d-object-detection alfred alfred-py deeplearning network pycocotools pytorch segmentation sensor-fusion video-combiner

Last synced: 31 Jul 2024

https://github.com/adaptivemotorcontrollab/cebra

Learnable latent embeddings for joint behavioral and neural analysis - Official implementation of CEBRA

contrastive-learning machine-learning neuroscience-methods pytorch

Last synced: 24 Sep 2024

https://github.com/nyoki-mtl/pytorch-EverybodyDanceNow

Implementation of Everybody Dance Now by pytorch

everybody-dance-now pytorch

Last synced: 01 Aug 2024

https://github.com/xiaoyufenfei/Efficient-Segmentation-Networks

Lightweight models for real-time semantic segmentationon PyTorch (include SQNet, LinkNet, SegNet, UNet, ENet, ERFNet, EDANet, ESPNet, ESPNetv2, LEDNet, ESNet, FSSNet, CGNet, DABNet, Fast-SCNN, ContextNet, FPENet, etc.)

camvid cityscapes computer-vision driving-scene-understanding efficient-segmentation-networks image-segmentation lightweight-semantic-segmentation neural-networks pytorch real-time-semantic-segmentation scene-understanding segmentation semantic-segmentation semantic-segmentation-models

Last synced: 31 Jul 2024

https://github.com/DataXujing/YOLO-v5

:art: Pytorch YOLO v5 训练自己的数据集超详细教程!!! :art: (提供PDF训练教程下载)

object-detection pytorch yolov5

Last synced: 02 Aug 2024

https://github.com/VITA-Group/EnlightenGAN

[IEEE TIP] "EnlightenGAN: Deep Light Enhancement without Paired Supervision" by Yifan Jiang, Xinyu Gong, Ding Liu, Yu Cheng, Chen Fang, Xiaohui Shen, Jianchao Yang, Pan Zhou, Zhangyang Wang

gan generative-adversarial-networks low-light low-light-enhance pytorch unsupervised-learning

Last synced: 01 Aug 2024

https://github.com/alexandre01/deepsvg

[NeurIPS 2020] Official code for the paper "DeepSVG: A Hierarchical Generative Network for Vector Graphics Animation". Includes a PyTorch library for deep learning with SVG data.

autoencoder deep-learning deep-svg library machine-learning python pytorch sketch-rnn sketches svg svg-animations svg-vae transformer

Last synced: 02 Aug 2024

https://github.com/NVIDIA/retinanet-examples

Fast and accurate object detection with end-to-end GPU optimization

deep-learning neural-network object-detection python pytorch retinanet tensorrt

Last synced: 07 Aug 2024

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

PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference

deep-learning pruning pytorch

Last synced: 01 Aug 2024

https://github.com/WenjieDu/PyPOTS

A Python toolkit/library for reality-centric machine/deep learning and data mining on partially-observed time series, including SOTA neural network models for scientific analysis tasks of imputation, classification, clustering, forecasting, & anomaly detection on incomplete industrial (irregularly-sampled) multivariate TS with NaN missing values

classification clustering data-mining data-science deep-learning forecasting healthcare imputation incomplete industrial interpolation machine-learning missing-values missingness neural-network partially-observed-time-series pytorch science-research time-series time-series-analysis

Last synced: 01 Aug 2024

https://github.com/moemen95/Pytorch-Project-Template

A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.

condense-net dcgan deep-learning dqn erfnet machine-learning pytorch pytorch-project-template pytorch-template pytorch-tutorial scemantic semantic-segmentation-models

Last synced: 03 Aug 2024

https://github.com/THUDM/SwissArmyTransformer

SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.

pretrained-models pytorch transformer

Last synced: 01 Aug 2024

https://github.com/switchablenorms/Switchable-Normalization

Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10779

convolutional-neural-networks deeplearning imagenet normalization pytorch

Last synced: 02 Aug 2024

https://github.com/VikParuchuri/zero_to_gpt

Go from no deep learning knowledge to implementing GPT.

deep-learning python pytorch tutorial

Last synced: 01 Aug 2024

https://github.com/moemen95/PyTorch-Project-Template

A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.

condense-net dcgan deep-learning dqn erfnet machine-learning pytorch pytorch-project-template pytorch-template pytorch-tutorial scemantic semantic-segmentation-models

Last synced: 01 Aug 2024

https://github.com/explosion/curated-transformers

🤖 A PyTorch library of curated Transformer models and their composable components

albert bert camembert dolly2 falcon gptneox llama llm llms nlp pytorch roberta transformer transformers xlm-roberta

Last synced: 24 Sep 2024

https://github.com/ZhiqingXiao/rl-book

Source codes for the book "Reinforcement Learning: Theory and Python Implementation"

deep-reinforcement-learning gym openai-gym python pytorch reinforcement-learning tensorflow tensorflow2

Last synced: 01 Aug 2024

https://github.com/frotms/paddleocr2pytorch

PaddleOCR inference in PyTorch. Converted from [PaddleOCR](https://github.com/PaddlePaddle/PaddleOCR)

crnn db multilingual-models ocr paddleocr paddlepaddle pytorch pytorchocr

Last synced: 25 Sep 2024

https://github.com/lonePatient/Bert-Multi-Label-Text-Classification

This repo contains a PyTorch implementation of a pretrained BERT model for multi-label text classification.

albert bert fine-tuning multi-label-classification nlp pytorch pytorch-implmention text-classification transformers xlnet

Last synced: 04 Aug 2024

https://github.com/philipperemy/n-beats

Keras/Pytorch implementation of N-BEATS: Neural basis expansion analysis for interpretable time series forecasting.

deep-learning neural-networks pytorch series-forecasting

Last synced: 03 Aug 2024

https://github.com/vahidk/tfrecord

Standalone TFRecord reader/writer with PyTorch data loaders

dataset loader pytorch tensorflow tfrecord

Last synced: 03 Aug 2024

https://github.com/airctic/icevision

An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come

ai annotation-parsers coco-dataset coco-parser computer-vision deep-learning effecientdet fastai faster-rcnn mask-rcnn object-detection pycocotools python pytorch pytorch-lightning tutorials voc-dataset voc-parser

Last synced: 03 Aug 2024

https://github.com/wasidennis/AdaptSegNet

Learning to Adapt Structured Output Space for Semantic Segmentation, CVPR 2018 (spotlight)

adversarial-learning computer-vision deep-learning domain-adaptation generative-adversarial-network pytorch semantic-segmentation

Last synced: 05 Aug 2024

https://github.com/zisianw/FaceBoxes.PyTorch

A PyTorch Implementation of FaceBoxes

face-detection faceboxes pytorch

Last synced: 31 Jul 2024

https://github.com/NVIDIA/modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

deep-learning machine-learning nvidia-gpu physics pytorch

Last synced: 31 Jul 2024