Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
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.
- GitHub: https://github.com/topics/pytorch
- Wikipedia: https://en.wikipedia.org/wiki/PyTorch
- Repo: https://github.com/pytorch/pytorch
- Created by: Facebook's AI Research lab (FAIR)
- Released: September 2016
- Related Topics: python, pytorch-tutorial,
- Last updated: 2024-11-10 00:23:23 UTC
- JSON Representation
https://github.com/bigmb/unet-segmentation-pytorch-nest-of-unets
Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet
imagesegmentation python3 pytorch segmentation tensorvision torch torchvision unet
Last synced: 14 Oct 2024
https://github.com/harry24k/adversarial-attacks-pytorch
PyTorch implementation of adversarial attacks [torchattacks]
adversarial-attacks deep-learning pytorch
Last synced: 09 Oct 2024
https://github.com/rlworkgroup/garage
A toolkit for reproducible reinforcement learning research.
pytorch reproducibility rl-algorithms tensorflow
Last synced: 14 Oct 2024
https://github.com/nobody132/masr
中文语音识别; Mandarin Automatic Speech Recognition;
chinese-speech-recognition mandarin-chinese pytorch speech-recognition
Last synced: 14 Oct 2024
https://github.com/ncullen93/torchsample
Train AI models efficiently on medical images using any framework
deep-learning keras medical-imaging neuroimaging pytorch
Last synced: 28 Aug 2024
https://github.com/nitrain/nitrain
Train AI models efficiently on medical images using any framework
deep-learning keras medical-imaging neuroimaging pytorch
Last synced: 13 Oct 2024
https://github.com/microsoft/deepspeed-mii
MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.
deep-learning inference pytorch
Last synced: 07 Oct 2024
https://github.com/curt-park/rainbow-is-all-you-need
Rainbow is all you need! A step-by-step tutorial from DQN to Rainbow
colab-notebook dqn gym-environment nbviewer pytorch rainbow reinforcement-learning
Last synced: 14 Oct 2024
https://github.com/open-mmlab/mmaction
An open-source toolbox for action understanding based on PyTorch
action-detection action-recognition pytorch spatial-temporal-action-detection temporal-action-detection temporal-action-localization video-understanding
Last synced: 28 Oct 2024
https://github.com/open-mmlab/mmrotate
OpenMMLab Rotated Object Detection Toolbox and Benchmark
detection openmmlab pytorch rotated-object
Last synced: 15 Oct 2024
https://github.com/adobe-research/custom-diffusion
Custom Diffusion: Multi-Concept Customization of Text-to-Image Diffusion (CVPR 2023)
computer-vision customization diffusion-models few-shot fine-tuning pytorch text-to-image-generation
Last synced: 14 Oct 2024
https://github.com/isl-org/Open3D-ML
An extension of Open3D to address 3D Machine Learning tasks
3d-object-detection 3d-perception datasets lidar object-detection pretrained-models pytorch rgbd semantic-segmentation tensorflow visualization
Last synced: 28 Oct 2024
https://github.com/IgorSusmelj/pytorch-styleguide
An unofficial styleguide and best practices summary for PyTorch
best-practices pytorch styleguide
Last synced: 02 Aug 2024
https://github.com/isl-org/open3d-ml
An extension of Open3D to address 3D Machine Learning tasks
3d-object-detection 3d-perception datasets lidar object-detection pretrained-models pytorch rgbd semantic-segmentation tensorflow visualization
Last synced: 15 Oct 2024
https://github.com/FunAudioLLM/SenseVoice
Multilingual Voice Understanding Model
ai aigc asr audio-event-classification cross-lingual gpt-4o llm multilingual python pytorch speech-emotion-recognition speech-recognition speech-to-text
Last synced: 03 Aug 2024
https://github.com/javierantoran/bayesian-neural-networks
Pytorch implementations of Bayes By Backprop, MC Dropout, SGLD, the Local Reparametrization Trick, KF-Laplace, SG-HMC and more
approximate-inference bayes-by-backprop bayesian-inference bayesian-neural-networks classification deep-learning hmc kronecker-factored-approximation langevin-dynamics local-reparametrization-trick mc-dropout mcmc out-of-distribution-detection pytorch regression reproducible-research sgld uncertainty uncertainty-neural-networks variational-inference
Last synced: 14 Oct 2024
https://github.com/pytorch/rl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
ai control decision-making distributed-computing machine-learning marl model-based-reinforcement-learning multi-agent-reinforcement-learning pytorch reinforcement-learning rl robotics torch
Last synced: 07 Oct 2024
https://github.com/rentruewang/koila
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.
deep-learning gradient-accumulation lazy-evaluation machine-learning memory-management neural-network out-of-memory python pytorch
Last synced: 02 Aug 2024
https://github.com/wolny/pytorch-3dunet
3D U-Net model for volumetric semantic segmentation written in pytorch
3d-segmentation 3d-unet dice-coefficient groupnorm pytorch pytorch-3dunet residual-unet semantic-segmentation unet unet-pytorch volumetric-data
Last synced: 14 Oct 2024
https://github.com/project-monai/tutorials
MONAI Tutorials
jupyter-notebook monai monai-tutorials monai-workflows pytorch
Last synced: 15 Oct 2024
https://github.com/uber/orbit
A Python package for Bayesian forecasting with object-oriented design and probabilistic models under the hood.
arima bayesian bayesian-methods bayesian-statistics changepoint exponential-smoothing forecast forecasting machine-learning orbit probabilistic probabilistic-programming pyro pystan python pytorch regression regression-models stan time-series
Last synced: 11 Oct 2024
https://github.com/clementchadebec/benchmark_vae
Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)
benchmarking beta-vae comparison normalizing-flows pixel-cnn pytorch reproducibility reproducible-research vae vae-gan vae-implementation vae-pytorch variational-autoencoder vq-vae wasserstein-autoencoder
Last synced: 15 Oct 2024
https://github.com/waleedka/hiddenlayer
Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.
deeplearning keras pytorch tensorboard tensorflow visualization
Last synced: 13 Oct 2024
https://github.com/Curt-Park/rainbow-is-all-you-need
Rainbow is all you need! A step-by-step tutorial from DQN to Rainbow
colab-notebook dqn gym-environment nbviewer pytorch rainbow reinforcement-learning
Last synced: 31 Oct 2024
https://github.com/haitongli/knowledge-distillation-pytorch
A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility
cifar10 computer-vision dark-knowledge deep-neural-networks knowledge-distillation model-compression pytorch
Last synced: 14 Oct 2024
https://github.com/uber/petastorm
Petastorm library enables single machine or distributed training and evaluation of deep learning models from datasets in Apache Parquet format. It supports ML frameworks such as Tensorflow, Pytorch, and PySpark and can be used from pure Python code.
deep-learning machine-learning parquet parquet-files pyarrow pyspark pytorch sysml tensorflow
Last synced: 15 Oct 2024
https://github.com/harskish/ganspace
Discovering Interpretable GAN Controls [NeurIPS 2020]
deep-learning gan generative-adversarial-network image-generation interactive-visualizations pytorch
Last synced: 15 Oct 2024
https://github.com/alibaba/EasyCV
An all-in-one toolkit for computer vision
classification computer-vision object-detection pytorch self-supervised-learning transformers vision-transformer
Last synced: 26 Oct 2024
https://github.com/yxlllc/DDSP-SVC
Real-time end-to-end singing voice conversion system based on DDSP (Differentiable Digital Signal Processing)
Last synced: 08 Nov 2024
https://github.com/cfzd/Ultra-Fast-Lane-Detection
Ultra Fast Structure-aware Deep Lane Detection (ECCV 2020)
autonomous-driving autonomous-vehicles cnn lane-detection lane-detector lane-finding pytorch road-detection
Last synced: 28 Oct 2024
https://github.com/alibaba/easycv
An all-in-one toolkit for computer vision
classification computer-vision object-detection pytorch self-supervised-learning transformers vision-transformer
Last synced: 14 Oct 2024
https://github.com/apple/ml-cvnets
CVNets: A library for training computer vision networks
ade20k classification computer-vision deep-learning detection imagenet machine-learning mscoco pascal-voc pytorch segmentation
Last synced: 07 Oct 2024
https://github.com/csailvision/gandissect
Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/
deep-learning gan generative-adversarial-network image-manipulation interactive-visualizations interpretable-ml pytorch
Last synced: 15 Oct 2024
https://github.com/shunsukesaito/PIFu
This repository contains the code for the paper "PIFu: Pixel-Aligned Implicit Function for High-Resolution Clothed Human Digitization"
3d-printing 3d-reconstruction arvr computer-graphics fashion geometry-processing human iccv2019 pifu pytorch
Last synced: 06 Nov 2024
https://github.com/CSAILVision/gandissect
Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/
deep-learning gan generative-adversarial-network image-manipulation interactive-visualizations interpretable-ml pytorch
Last synced: 02 Aug 2024
https://github.com/continualai/avalanche
Avalanche: an End-to-End Library for Continual Learning based on PyTorch.
benchmarks continual-learning continualai deep-learning evaluation framework library lifelong-learning metrics pytorch strategies training
Last synced: 15 Oct 2024
https://github.com/hkchengrex/XMem
[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model
computer-vision deep-learning eccv-2022 eccv2022 pytorch segmentation video-object-segmentation video-segmentation
Last synced: 09 Nov 2024
https://github.com/JavierAntoran/Bayesian-Neural-Networks
Pytorch implementations of Bayes By Backprop, MC Dropout, SGLD, the Local Reparametrization Trick, KF-Laplace, SG-HMC and more
approximate-inference bayes-by-backprop bayesian-inference bayesian-neural-networks classification deep-learning hmc kronecker-factored-approximation langevin-dynamics local-reparametrization-trick mc-dropout mcmc out-of-distribution-detection pytorch regression reproducible-research sgld uncertainty uncertainty-neural-networks variational-inference
Last synced: 02 Aug 2024
https://github.com/swz30/restormer
[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.
cvpr2022 defocus-deblurring efficient-transformers high-resolution image-deblurring image-deraining image-restoration low-level-vision motion-deblurring pytorch transformer
Last synced: 14 Oct 2024
https://github.com/swz30/Restormer
[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.
cvpr2022 defocus-deblurring efficient-transformers high-resolution image-deblurring image-deraining image-restoration low-level-vision motion-deblurring pytorch transformer
Last synced: 28 Oct 2024
https://github.com/davda54/sam
SAM: Sharpness-Aware Minimization (PyTorch)
optimizer pytorch sam sharpness-aware
Last synced: 09 Oct 2024
https://github.com/hkchengrex/xmem
[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model
computer-vision deep-learning eccv-2022 eccv2022 pytorch segmentation video-object-segmentation video-segmentation
Last synced: 15 Oct 2024
https://github.com/wkentaro/pytorch-fcn
PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)
computer-vision convolutional-networks deep-learning fcn fcn8s pytorch semantic-segmentation
Last synced: 25 Sep 2024
https://github.com/facebookresearch/theseus
A library for differentiable nonlinear optimization
bilevel-optimization computer-vision deep-learning differentiable-optimization embodied-ai gauss-newton implicit-differentiation levenberg-marquardt nonlinear-least-squares pytorch robotics
Last synced: 15 Oct 2024
https://github.com/clementchadebec/benchmark_VAE
Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)
benchmarking beta-vae comparison normalizing-flows pixel-cnn pytorch reproducibility reproducible-research vae vae-gan vae-implementation vae-pytorch variational-autoencoder vq-vae wasserstein-autoencoder
Last synced: 02 Aug 2024
https://github.com/thu-coai/CDial-GPT
A Large-scale Chinese Short-Text Conversation Dataset and Chinese pre-training dialog models
dialogue gpt gpt-2 lccc pytorch text-generation
Last synced: 03 Aug 2024
https://github.com/huggingface/transfer-learning-conv-ai
🦄 State-of-the-Art Conversational AI with Transfer Learning
chatbots deep-learning dialog gpt gpt-2 neural-networks nlp pytorch transfer-learning
Last synced: 15 Oct 2024
https://github.com/Lightning-Universe/lightning-flash
Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
classification deep-learning fiftyone icevision machine-learning object-detection open3d pytorch pytorch-lightning pytorch-video tabular-data tasks-flash torch-geometric
Last synced: 10 Nov 2024
https://github.com/PyTorchLightning/lightning-flash
Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
classification deep-learning fiftyone icevision machine-learning object-detection open3d pytorch pytorch-lightning pytorch-video tabular-data tasks-flash torch-geometric
Last synced: 08 Nov 2024
https://github.com/gram-ai/capsule-networks
A PyTorch implementation of the NIPS 2017 paper "Dynamic Routing Between Capsules".
capsule-networks machine-learning mnist pytorch
Last synced: 29 Oct 2024
https://github.com/lightning-universe/lightning-flash
Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
classification deep-learning fiftyone icevision machine-learning object-detection open3d pytorch pytorch-lightning pytorch-video tabular-data tasks-flash torch-geometric
Last synced: 26 Sep 2024
https://shunsukesaito.github.io/PIFu/
This repository contains the code for the paper "PIFu: Pixel-Aligned Implicit Function for High-Resolution Clothed Human Digitization"
3d-printing 3d-reconstruction arvr computer-graphics fashion geometry-processing human iccv2019 pifu pytorch
Last synced: 03 Aug 2024
https://github.com/shunsukesaito/pifu
This repository contains the code for the paper "PIFu: Pixel-Aligned Implicit Function for High-Resolution Clothed Human Digitization"
3d-printing 3d-reconstruction arvr computer-graphics fashion geometry-processing human iccv2019 pifu pytorch
Last synced: 14 Oct 2024
https://github.com/svip-lab/impersonator
PyTorch implementation of our ICCV 2019 paper: Liquid Warping GAN: A Unified Framework for Human Motion Imitation, Appearance Transfer and Novel View Synthesis
Last synced: 14 Oct 2024
https://github.com/deepset-ai/farm
:house_with_garden: Fast & easy transfer learning for NLP. Harvesting language models for the industry. Focus on Question Answering.
bert deep-learning germanbert language-models ner nlp nlp-framework nlp-library pretrained-models pytorch question-answering roberta transfer-learning xlnet-pytorch
Last synced: 15 Oct 2024
https://github.com/deepset-ai/FARM
:house_with_garden: Fast & easy transfer learning for NLP. Harvesting language models for the industry. Focus on Question Answering.
bert deep-learning germanbert language-models ner nlp nlp-framework nlp-library pretrained-models pytorch question-answering roberta transfer-learning xlnet-pytorch
Last synced: 04 Nov 2024
https://github.com/unslothai/hyperlearn
2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.
data-analysis data-science deep-learning econometrics gpu machine-learning neural-network optimization python pytorch regression-models research scikit-learn statistics statsmodels tensor
Last synced: 13 Oct 2024
https://github.com/ContinualAI/avalanche
Avalanche: an End-to-End Library for Continual Learning based on PyTorch.
benchmarks continual-learning continualai deep-learning evaluation framework library lifelong-learning metrics pytorch strategies training
Last synced: 02 Aug 2024
https://github.com/Project-MONAI/tutorials
MONAI Tutorials
jupyter-notebook monai monai-tutorials monai-workflows pytorch
Last synced: 02 Aug 2024
https://github.com/thudm/cogdl
CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)
gnn-model graph-classification graph-embedding graph-neural-networks leaderboard link-prediction node-classification pytorch
Last synced: 15 Oct 2024
https://github.com/djdefrag/qualityscaler
QualityScaler - image/video AI upscaler app
amd anime compression-artifact-reduction deep-learning directx-12 gui-application intel manga noise-reduction nvidia opencv python python3 pytorch pytorch-directml super-resolution video video-processing windows
Last synced: 15 Oct 2024
https://github.com/Djdefrag/QualityScaler
QualityScaler - image/video AI upscaler app
amd anime compression-artifact-reduction deep-learning directx-12 gui-application intel manga noise-reduction nvidia opencv python python3 pytorch pytorch-directml super-resolution video video-processing windows
Last synced: 29 Oct 2024
https://github.com/THUDM/CogView
Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".
pretrained-models pytorch text-to-image transformers
Last synced: 08 Nov 2024
https://github.com/black0017/medicalzoopytorch
A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation
3d-convolutional-network brats2018 brats2019 deep-learning densenet iseg iseg-challenge medical-image-processing medical-image-segmentation medical-imaging mrbrains18 pytorch resnet segmentation segmentation-models unet unet-image-segmentation
Last synced: 15 Oct 2024
https://github.com/tugstugi/dl-colab-notebooks
Try out deep learning models online on Google Colab
cnn colab-notebook deep-learning deep-neural-networks google-colab google-colaboratory jupyter-notebook jupyter-notebooks pytorch tensorflow
Last synced: 14 Oct 2024
https://github.com/THUDM/cogdl
CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)
gnn-model graph-classification graph-embedding graph-neural-networks leaderboard link-prediction node-classification pytorch
Last synced: 07 Nov 2024
https://github.com/THUDM/CogDL
CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)
gnn-model graph-classification graph-embedding graph-neural-networks leaderboard link-prediction node-classification pytorch
Last synced: 02 Aug 2024
https://github.com/ZijunDeng/pytorch-semantic-segmentation
PyTorch for Semantic Segmentation
deep-learning fully-convolutional-networks pytorch semantic-segmentation
Last synced: 03 Aug 2024
https://github.com/longcw/faster_rcnn_pytorch
Faster RCNN with PyTorch
computer-vision detection faster-rcnn pytorch
Last synced: 05 Nov 2024
https://github.com/wxinlong/solo
SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
instance-segmentation object-detection pytorch solo solov2
Last synced: 14 Oct 2024
https://github.com/thudm/cogview
Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".
pretrained-models pytorch text-to-image transformers
Last synced: 14 Oct 2024
https://github.com/Lightning-Universe/lightning-bolts
Toolbox of models, callbacks, and datasets for AI/ML researchers.
ai gan image-processing machine-learning natural-language-processing pytorch supervised-learning
Last synced: 06 Nov 2024
https://github.com/pytorch/opacus
Training PyTorch models with differential privacy
deep-learning differential-privacy machine-learning neural-network privacy-preserving-machine-learning pytorch
Last synced: 29 Oct 2024
https://github.com/bearpaw/pytorch-classification
Classification with PyTorch.
cifar10 cifar100 classification densenet imagenet preresnet pytorch resnet resnext wide-residual-networks wrn
Last synced: 14 Oct 2024
https://github.com/argusswift/yolov4-pytorch
This is a pytorch repository of YOLOv4, attentive YOLOv4 and mobilenet YOLOv4 with PASCAL VOC and COCO
attention cbam mobilenetv2 mobilenetv3 object-detection pytorch senet yolov4
Last synced: 14 Oct 2024
https://github.com/argusswift/YOLOv4-pytorch
This is a pytorch repository of YOLOv4, attentive YOLOv4 and mobilenet YOLOv4 with PASCAL VOC and COCO
attention cbam mobilenetv2 mobilenetv3 object-detection pytorch senet yolov4
Last synced: 09 Nov 2024
https://github.com/nikhilbarhate99/ppo-pytorch
Minimal implementation of clipped objective Proximal Policy Optimization (PPO) in PyTorch
deep-learning deep-reinforcement-learning policy-gradient ppo ppo-pytorch proximal-policy-optimization pytorch pytorch-implmention pytorch-tutorial reinforcement-learning reinforcement-learning-algorithms
Last synced: 14 Oct 2024
https://github.com/graph4ai/graph4nlp
Graph4nlp is the library for the easy use of Graph Neural Networks for NLP. Welcome to visit our DLG4NLP website (https://dlg4nlp.github.io/index.html) for various learning resources!
deep-learning graph-neural-networks machine-learning natural-language-processing nlp pytorch
Last synced: 15 Oct 2024
https://github.com/median-research-group/LibMTL
A PyTorch Library for Multi-Task Learning
deep-learning mmoe mtl multi-domain-learning multi-objective-optimization multi-task-learning multiobjective-optimization multitask-learning ple python pytorch
Last synced: 28 Oct 2024
https://github.com/median-research-group/libmtl
A PyTorch Library for Multi-Task Learning
deep-learning mmoe mtl multi-domain-learning multi-objective-optimization multi-task-learning multiobjective-optimization multitask-learning ple python pytorch
Last synced: 14 Oct 2024
https://github.com/pytorch/tnt
A lightweight library for PyTorch training tools and utilities
deep-learning machine-learning neural-network python pytorch reinforcement-learning
Last synced: 07 Oct 2024
https://github.com/vahidk/effectivepytorch
PyTorch tutorials and best practices.
deep-learning ebook machine-learning neural-network pytorch
Last synced: 15 Oct 2024
https://github.com/vahidk/EffectivePyTorch
PyTorch tutorials and best practices.
deep-learning ebook machine-learning neural-network pytorch
Last synced: 27 Oct 2024
https://github.com/ymcui/Chinese-XLNet
Pre-Trained Chinese XLNet(中文XLNet预训练模型)
natural-language-processing nlp pytorch tensorflow xlnet
Last synced: 31 Oct 2024
https://github.com/ymcui/chinese-xlnet
Pre-Trained Chinese XLNet(中文XLNet预训练模型)
natural-language-processing nlp pytorch tensorflow xlnet
Last synced: 28 Oct 2024
https://github.com/lightning-universe/lightning-bolts
Toolbox of models, callbacks, and datasets for AI/ML researchers.
ai gan image-processing machine-learning natural-language-processing pytorch supervised-learning
Last synced: 15 Oct 2024
https://github.com/ai-forever/ru-dalle
Generate images from texts. In Russian
dalle image-generation openai python pytorch russian russian-language text-to-image transformer
Last synced: 09 Oct 2024
https://github.com/huggingface/huggingface_hub
The official Python client for the Huggingface Hub.
deep-learning hacktoberfest machine-learning model-hub models natural-language-processing pretrained-models pytorch
Last synced: 22 Oct 2024
https://github.com/vita-group/transgan
[NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang
gan pytorch transformer transformer-encoder transformer-models
Last synced: 14 Oct 2024
https://github.com/nvlabs/pwc-net
PWC-Net: CNNs for Optical Flow Using Pyramid, Warping, and Cost Volume, CVPR 2018 (Oral)
caffe computer-vision cvpr2018 deeplearning optical-flow pwc-net pytorch
Last synced: 15 Oct 2024
https://github.com/VITA-Group/TransGAN
[NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang
gan pytorch transformer transformer-encoder transformer-models
Last synced: 03 Aug 2024
https://github.com/LoSealL/VideoSuperResolution
A collection of state-of-the-art video or single-image super-resolution architectures, reimplemented in tensorflow.
carn dbpn dncnn edsr frvsr ntire2019 pytorch rcan rdn srcnn srgan srmd super-resolution tensorflow vdsr vespcn vsr
Last synced: 28 Oct 2024
https://github.com/black0017/MedicalZooPytorch
A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation
3d-convolutional-network brats2018 brats2019 deep-learning densenet iseg iseg-challenge medical-image-processing medical-image-segmentation medical-imaging mrbrains18 pytorch resnet segmentation segmentation-models unet unet-image-segmentation
Last synced: 03 Aug 2024
https://github.com/yuliangxiu/icon
[CVPR'22] ICON: Implicit Clothed humans Obtained from Normals
3d-reconstruction animation avatar-generator cloth-simulation computer-graphics computer-vision human-pose-estimation implicit-functions mesh-deformation metaverse normal-maps pifu pifuhd pose-estimation pytorch smpl smpl-body smpl-model smplx virtual-humans
Last synced: 14 Oct 2024
https://github.com/dusty-nv/jetson-containers
Machine Learning Containers for NVIDIA Jetson and JetPack-L4T
containers docker dockerfiles jetson machine-learning numpy nvidia pandas pytorch ros-containers ros2-foxy scikit-learn tensorflow
Last synced: 13 Oct 2024
https://github.com/airaria/textbrewer
A PyTorch-based knowledge distillation toolkit for natural language processing
bert distillation knowledge nlp pytorch
Last synced: 09 Oct 2024
https://github.com/airaria/TextBrewer
A PyTorch-based knowledge distillation toolkit for natural language processing
bert distillation knowledge nlp pytorch
Last synced: 03 Nov 2024
https://github.com/kakaobrain/fast-autoaugment
Official Implementation of 'Fast AutoAugment' in PyTorch.
augmentation automated-machine-learning automl cnn computer-vision convolutional-neural-networks deep-learning distributed image-classification pytorch
Last synced: 29 Oct 2024
https://github.com/microsoft/EdgeML
This repository provides code for machine learning algorithms for edge devices developed at Microsoft Research India.
bonsai classifier cpp deep-learning edge-computing edge-devices edge-machine-learning emi-rnn fastgrnn fastrnn iot-device machine-learning machine-learning-algorithms microsoft microsoft-research protonn pytorch resource-constrained-ml sensor tensorflow
Last synced: 01 Nov 2024