yichuan9527

ray

Ray is a unified framework for scaling AI and Python applications. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

Dive-into-DL-PyTorch

本项目将《动手学深度学习》(Dive into Deep Learning)原书中的MXNet实现改为PyTorch实现。

kornia

Geometric Computer Vision Library for Spatial AI

Deep-Reinforcement-Learning-Algorithms-with-PyTorch

PyTorch implementations of deep reinforcement learning algorithms and environments

captum

Model interpretability and understanding for PyTorch

face-parsing.PyTorch

Using modified BiSeNet for face parsing in PyTorch

torchcv

TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision

AutoDL-Projects

Automated deep learning algorithms implemented in PyTorch.

AutoAugment

Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow

ATSS

Bridging the Gap Between Anchor-based and Anchor-free Detection via Adaptive Training Sample Selection, CVPR, Oral, 2020

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.)

SiamDW

[CVPR'19 Oral] Deeper and Wider Siamese Networks for Real-Time Visual Tracking

meta-transfer-learning

TensorFlow and PyTorch implementation of "Meta-Transfer Learning for Few-Shot Learning" (CVPR2019)

hgcn

Hyperbolic Graph Convolutional Networks in PyTorch.

visual-machine-learning-notes

Machine Learning Conference & Summer School Notes. 🦄

person_search_demo

利用YOLOv3结合行人重识别模型，实现行人的检测识别，查找特定行人

Tricks-of-Semi-supervisedDeepLeanring-Pytorch

PseudoLabel 2013, VAT, PI model, Tempens, MeanTeacher, ICT, MixMatch, FixMatch

PyTorch-Deep-Learning-Template

A Pytorch Computer Vision template to quick start your next project! 🚀🚀

Cross-Modal-Re-ID-baseline

Pytorch Code for Cross-Modality Person Re-Identification (Visible Thermal/Infrared Re-ID)

PyTorch-Image-Retrieval

A PyTorch framework for an image retrieval task including implementation of N-pair Loss (NIPS 2016) and Angular Loss (ICCV 2017).

pytorch-maml

An Implementation of Model-Agnostic Meta-Learning in PyTorch with Torchmeta

radish

C++ model train&inference framework

ViP

Video Platform for Action Recognition and Object Detection in Pytorch

Awesome-Person-Re-Identification

Awesome Person Re-Identification

SV-X-Softmax

Occluded-DukeMTMC-Dataset

This is the Occluded-DukeMTMC dataset from the ICCV2019 paper "Pose-Guided Feature Alignment for Occluded Person Re-Identification"

ShuffleBN

A ShuffleBatchNorm layer to shuffle BatchNorm statistics across multiple GPUs

NASBench-PyTorch

A PyTorch implementation of NASBench

ssl-suite

SSL using PyTorch

Learning-to-impute

Learning to Impute: A General Framework for Semi-supervised Learning