-
PyTorch implementation of over 30 realtime semantic segmentations models, e.g. BiSeNetv1, BiSeNetv2, CGNet, ContextNet, DABNet, DDRNet, EDANet, ENet, ERFNet, ESPNet, ESPNetv2, FastSCNN, ICNet, LEDN…
-
simple-yolo-pytorch Public
Simplified PyTorch implementation of YOLO, support multi-gpu training and validating, automatic mixed precision training etc.
-
medical-segmentation-pytorch Public
PyTorch implementation of medical semantic segmentations models, e.g. UNet, UNet++, DUCKNet, ResUNet, ResUNet++, and support knowledge distillation, distributed training, Optuna etc.
-
image-classification-pytorch Public
Simplified PyTorch implementation of image classification, support CIFAR10, CIFAR100, MNIST, custom dataset, multi-gpu training and validating, automatic mixed precision training, knowledge distill…
-
audio-classification-pytorch Public
Simplified PyTorch implementation of audio classification, support multi-gpu training and validating, automatic mixed precision training, knowledge distillation etc.
-
PyTorch implementation of efficient image super-resolution models, e.g. SRCNN, ESPCN, FSRCNN, DRCN, VDSR, DRRN, EDSR, LapSRN, IDN, CARN etc.
{{ message }}