PyTorch | Ecosyste.ms: Awesome

https://github.com/autodistill/autodistill

Images to inference with no labeling (use foundation models to train supervised models).

auto-labeling computer-vision deep-learning foundation-models grounding-dino image-annotation image-classification instance-segmentation labeling-tool machine-learning model-distillation multimodal object-detection pytorch segment-anything yolov5 yolov8

Last synced: 15 Oct 2024

https://github.com/po-hsun-su/pytorch-ssim

pytorch structural similarity (SSIM) loss

image-analysis image-processing pytorch

Last synced: 15 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: 30 Oct 2024

https://github.com/labmlai/labml

🔎 Monitor deep learning model training and hardware usage from your mobile phone 📱

analytics deep-learning experiment fastai keras keras-tensorflow machine-learning mobile pytorch pytorch-lightning tensorboard tensorflow tensorflow2 visualization

Last synced: 13 Oct 2024

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/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: 11 Nov 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/Project-MONAI/tutorials

MONAI Tutorials

jupyter-notebook monai monai-tutorials monai-workflows pytorch

Last synced: 12 Nov 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/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: 12 Nov 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: 12 Nov 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: 11 Nov 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/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/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/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/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: 13 Nov 2024

https://github.com/yxlllc/DDSP-SVC

Real-time end-to-end singing voice conversion system based on DDSP (Differentiable Digital Signal Processing)

pytorch

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: 12 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: 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/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/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/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/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: 12 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://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/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://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/apachecn/.github

ApacheCN 开源组织：公告、介绍、成员、活动、交流方式

dl ml python pytorch solidity spark

Last synced: 12 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

gan pose pytorch

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/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/ZijunDeng/pytorch-semantic-segmentation

PyTorch for Semantic Segmentation

deep-learning fully-convolutional-networks pytorch semantic-segmentation

Last synced: 14 Nov 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/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: 11 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: 07 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/longcw/faster_rcnn_pytorch

Faster RCNN with PyTorch

computer-vision detection faster-rcnn pytorch

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