xiaolonghao Goto Github PK

pytorch-cyclegan-and-pix2pix

Image-to-image translation in PyTorch (e.g., horse2zebra, edges2cats, and more)

pytorch-faster-rcnn

pytorch1.0 updated. Support cpu test and demo.

pytorch-grad-cam

Many Class Activation Map methods implemented in Pytorch for CNNs and Vision Transformers. Including Grad-CAM, Grad-CAM++, Score-CAM, Ablation-CAM and XGrad-CAM

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

pytorch-mask-rcnn

pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.

pytorch-unet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

pytorch-yolo-v3

A PyTorch implementation of the YOLO v3 object detection algorithm

pytorch-yolov3

Minimal PyTorch implementation of YOLOv3

pytorch-yolov4

Minimal PyTorch implementation of YOLOv4

random-erasing

Random Erasing Data Augmentation. Experiments on CIFAR10, CIFAR100 and Fashion-MNIST

real-world-masked-face-dataset

Real-World Masked Face Dataset，口罩人脸数据集

refinenet-pytorch

RefineNet-101 VOC in PyTorch

reid-mgn

Simple pytorch implement of paper: Learning Discriminative Features with Multiple Granularities for Person Re-Identification

reinforcement-learning-algorithms

This repository contains most of pytorch implementation based classic deep reinforcement learning algorithms, including - DQN, DDQN, Dueling Network, DDPG, SAC, A2C, PPO, TRPO. (More algorithms are still in progress)

reinforcement_learning_projects

Using (deep) reinforcement_learning algorithm to practice on OpenAI Gym, Unity ML-Agents,and other virtual environments. Using Python ,Pytorch

rl-adventure-2

PyTorch0.4 implementation of: actor critic / proximal policy optimization / acer / ddpg / twin dueling ddpg / soft actor critic / generative adversarial imitation learning / hindsight experience replay

segmentation_models.pytorch

Segmentation models with pretrained backbones. PyTorch.

simple-faster-rcnn-copy

整理simple-faster-rcnn的环境

simple-faster-rcnn-pytorch

A simplified implemention of Faster R-CNN that replicate performance from origin paper

smart_construction

Head Person Helmet Detection on Construction Sites，基于目标检测工地安全帽和禁入危险区域识别系统，🚀😆附 YOLOv5 训练自己的数据集超详细教程🚀😆2021.3新增可视化界面❗❗

spanet

Spatial Attentive Single-Image Deraining with a High Quality Real Rain Dataset (CVPR'19)

ssd

High quality, fast, modular reference implementation of SSD in PyTorch

ssd.pytorch

A PyTorch Implementation of Single Shot MultiBox Detector

ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.

stargan

Official PyTorch Implementation of StarGAN - CVPR 2018

super-mario-bros-a3c-pytorch

Asynchronous Advantage Actor-Critic (A3C) algorithm for Super Mario Bros

super-mario-bros-ppo-pytorch

Proximal Policy Optimization (PPO) algorithm for Super Mario Bros

textrecognitiondatagenerator

A synthetic data generator for text recognition

tianshou

An elegant, flexible, and superfast PyTorch deep reinforcement learning platform.