Pinned Repositories
efficient-image-super-resolution-pytorch
PyTorch implementation of efficient image super-resolution models, e.g. SRCNN, ESPCN, FSRCNN, DRCN, VDSR, DRRN, EDSR, LapSRN, IDN, CARN etc.
image-classification-pytorch
Simplified PyTorch implementation of image classification, support CIFAR10, CIFAR100, MNIST, custom dataset, multi-gpu training and validating, automatic mixed precision training, knowledge distillation, hyperparameter optimization using Optuna etc.
realtime-semantic-segmentation-pytorch
PyTorch implementation of over 30 realtime semantic segmentations models, e.g. BiSeNetv1, BiSeNetv2, CGNet, ContextNet, DABNet, DDRNet, EDANet, ENet, ERFNet, ESPNet, ESPNetv2, FastSCNN, ICNet, LEDNet, LinkNet, PP-LiteSeg, SegNet, ShelfNet, STDC, SwiftNet, and support knowledge distillation, distributed training, Optuna etc.
simple-yolo-pytorch
Simplified PyTorch implementation of YOLO, support multi-gpu training and validating, automatic mixed precision training etc.
zh320's Repositories
zh320/realtime-semantic-segmentation-pytorch
PyTorch implementation of over 30 realtime semantic segmentations models, e.g. BiSeNetv1, BiSeNetv2, CGNet, ContextNet, DABNet, DDRNet, EDANet, ENet, ERFNet, ESPNet, ESPNetv2, FastSCNN, ICNet, LEDNet, LinkNet, PP-LiteSeg, SegNet, ShelfNet, STDC, SwiftNet, and support knowledge distillation, distributed training, Optuna etc.
zh320/efficient-image-super-resolution-pytorch
PyTorch implementation of efficient image super-resolution models, e.g. SRCNN, ESPCN, FSRCNN, DRCN, VDSR, DRRN, EDSR, LapSRN, IDN, CARN etc.
zh320/image-classification-pytorch
Simplified PyTorch implementation of image classification, support CIFAR10, CIFAR100, MNIST, custom dataset, multi-gpu training and validating, automatic mixed precision training, knowledge distillation, hyperparameter optimization using Optuna etc.
zh320/simple-yolo-pytorch
Simplified PyTorch implementation of YOLO, support multi-gpu training and validating, automatic mixed precision training etc.