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/vainf/pytorch-msssim

Fast and differentiable MS-SSIM and SSIM for pytorch.

image-quality ms-ssim pytorch ssim structure-similarity

Last synced: 24 Sep 2024

https://github.com/carefree0910/MachineLearning

Machine learning algorithms implemented by pure numpy

deep-learning machine-learning numpy pytorch tensorflow visualization

Last synced: 02 Aug 2024

https://github.com/chenyuntc/PyTorchText

1st Place Solution for Zhihu Machine Learning Challenge . Implementation of various text-classification models.(知乎看山杯第一名解决方案)

fasttext lstm nlp pytorch textcnn textrcnn textrnn

Last synced: 31 Jul 2024

https://github.com/TorchEnsemble-Community/Ensemble-Pytorch

A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.

bagging deep-learning deeplearning ensemble ensemble-learning gradient-boosting neural-networks pytorch pytorch-tutorial voting-classifier

Last synced: 03 Aug 2024

https://github.com/naoto0804/pytorch-AdaIN

Unofficial pytorch implementation of 'Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization' [Huang+, ICCV2017]

cnn instance-normalization pytorch style-transfer

Last synced: 03 Aug 2024

https://github.com/xl0/lovely-tensors

Tensors, ready for human consumption

deep-learning library pytorch statistics visualization

Last synced: 24 Sep 2024

https://github.com/ternaus/TernausNet

UNet model with VGG11 encoder pre-trained on Kaggle Carvana dataset

image-segmentation pytorch

Last synced: 07 Aug 2024

https://github.com/ternaus/ternausnet

UNet model with VGG11 encoder pre-trained on Kaggle Carvana dataset

image-segmentation pytorch

Last synced: 24 Sep 2024

https://github.com/shi-labs/neighborhood-attention-transformer

Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022

neighborhood-attention pytorch

Last synced: 25 Sep 2024

https://github.com/caoscott/srec

PyTorch Implementation of "Lossless Image Compression through Super-Resolution"

compression lossless neural-network pytorch

Last synced: 24 Sep 2024

https://github.com/Res2Net/Res2Net-PretrainedModels

(ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"

backbone jittor multi-scale pytorch res2net

Last synced: 02 Aug 2024

https://github.com/cardwing/codes-for-lane-detection

Learning Lightweight Lane Detection CNNs by Self Attention Distillation (ICCV 2019)

cnn deep-learning lane-detection lua pytorch tensorflow

Last synced: 25 Sep 2024

https://github.com/caoscott/SReC

PyTorch Implementation of "Lossless Image Compression through Super-Resolution"

compression lossless neural-network pytorch

Last synced: 05 Aug 2024

https://github.com/eduardoleao052/js-pytorch

A JavaScript library like PyTorch, with GPU acceleration.

automatic-differentiation deep-learning javascript-library neural-networks pytorch

Last synced: 25 Sep 2024

https://github.com/maum-ai/voicefilter

Unofficial PyTorch implementation of Google AI's VoiceFilter system

audio-separation pytorch source-separation speech-separation voicefilter

Last synced: 25 Sep 2024

https://github.com/piddnad/ddcolor

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

computer-vision image-colorization pytorch

Last synced: 24 Sep 2024

https://github.com/cardwing/Codes-for-Lane-Detection

Learning Lightweight Lane Detection CNNs by Self Attention Distillation (ICCV 2019)

cnn deep-learning lane-detection lua pytorch tensorflow

Last synced: 30 Jul 2024

https://github.com/SHI-Labs/Neighborhood-Attention-Transformer

Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022

neighborhood-attention pytorch

Last synced: 01 Aug 2024

https://github.com/huggingface/pytorch-pretrained-biggan

🦋A PyTorch implementation of BigGAN with pretrained weights and conversion scripts.

artificial-intelligence biggan computer-vision gan generative-adversarial-network neural-network pytorch

Last synced: 24 Sep 2024

https://github.com/unum-cloud/uform

Pocket-Sized Multimodal AI for content understanding and generation across multilingual texts, images, and 🔜 video, up to 5x faster than OpenAI CLIP and LLaVA 🖼️ & 🖋️

bert clip clustering contrastive-learning cross-attention huggingface-transformers image-search language-vision llava multi-lingual multimodal neural-network openai openclip pretrained-models pytorch representation-learning semantic-search transformer vector-search

Last synced: 24 Sep 2024

https://github.com/huggingface/pytorch-pretrained-BigGAN

🦋A PyTorch implementation of BigGAN with pretrained weights and conversion scripts.

artificial-intelligence biggan computer-vision gan generative-adversarial-network neural-network pytorch

Last synced: 07 Aug 2024

https://github.com/shawn1993/cnn-text-classification-pytorch

CNNs for Sentence Classification in PyTorch

cnn-model pytorch

Last synced: 24 Sep 2024

https://github.com/descriptinc/descript-audio-codec

State-of-the-art audio codec with 90x compression factor. Supports 44.1kHz, 24kHz, and 16kHz mono/stereo audio.

audio audio-compression codec compression-algorithm deep-learning gans generative-adversarial-network pytorch residual-vector-quantization

Last synced: 05 Aug 2024

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/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/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/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/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/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/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/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/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/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: 26 Sep 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/atong01/conditional-flow-matching

TorchCFM: a Conditional Flow Matching library

continuous-normalizing-flows optimal-transport pytorch

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