Tensors and Dynamic neural networks in Python with strong GPU acceleration

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

deep learning for image processing including classification and object-detection etc.

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

PyTorch implementations of Generative Adversarial Networks.

Datasets, Transforms and Models specific to Computer Vision

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

🎨 Python Echarts Plotting Library

A list of all named GANs!

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

A Collection of Variational Autoencoders (VAE) in PyTorch.

Code release for ConvNeXt model

PyTorch to Keras model convertor

PyTorch implementation of neural style transfer algorithm

整理 pytorch 单机多 GPU 训练方法与原理

liver segmentation using deep learning

Demos for deep learning

visualization：filter、feature map、attention map、image-mask、grad-cam、human keypoint、guided-backpro

Prior Attention Network for Multi-Lesion Segmentation in Medical Images

To study pytorch can improve myself about deeplearing, so it is excited to create the repository. 为了更够更好地学习基于框架pytorch的深度学习，特地创建该仓库，用于记录如何通过pytorch完成深度学习的任务（如：文本分类、图像分类、语义分割、实例分割、目标检测等），同时也会不断更新一些像…

A demo to learn to use UNet-2D to solve the liver segmentation task.