5 results found Sort:

wang-xinyu
tensorrtx wang-xinyu
1.8k
7.0k
mit
104
Implementation of popular deep learning networks with TensorRT network definition API
crnn detr resnet yolo11 yolov3 yolov4 yolov5 yolov7 yolov8 yolov9 arcface mnasnet tensorrt retinaface squeezenet yolov3-spp mobilenetv2 mobilenetv3 swin-transformer
Created 2019-11-25
467 commits to master branch, last one about a month ago
rwightman
gen-efficientnet-pytorch rwightman
214
1.6k
apache-2.0
43
Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS
onnx fbnet caffe2 mnasnet pytorch mobilenetv3 efficientnet pretrained-models
Created 2019-05-11
109 commits to master branch, last one 5 months ago
Kedreamix
Pytorch-Image-Classification Kedreamix
24
202
unknown
3
用于pytorch的图像分类，包含多种模型方法，比如AlexNet，VGG，GoogleNet，ResNet，DenseNet等等，包含可完整运行的代码。除此之外，也有colab的在线运行代码，可以直接在colab在线运行查看结果。也可以迁移到自己的数据集进行迁移学习。
vgg resnet alexnet cifar10 mnasnet pytorch densenet goolenet inception mobilenet shufflenet
Created 2021-09-08
141 commits to main branch, last one 11 months ago
ZJCV
ZCls ZJCV
23
140
apache-2.0
5
Object Classification Training Framework
lmdb zcls acnet gcnet repvgg mnasnet pytorch ghostnet resnet3d groupnorm mpdataset non-local prefetcher cutmix-mixup resnet-resnext shufflenetv1-v2 gradient-clipping mobilenetv1-v2-v3 senet-sknet-resnest diverse-branch-block
This repository has been archived (exclude archived)
Created 2020-11-10
607 commits to master branch, last one 2 years ago
yang-ruixin
PyTorch-Image-Models-Multi-Label-Classification yang-ruixin
13
111
apache-2.0
3
Multi-label classification based on timm.
timm hrnet mixnet regnet resnet mnasnet pytorch densenet multi-task mobilenetv3 multi-label efficientnet pretrained-models transfer-learning pretrained-weights imagenet-classifier multi-task-learning gradient-centralization vision-transformer-models multi-label-classification
Created 2021-03-19
27 commits to main branch, last one 3 years ago