5 results found Sort:

wang-xinyu
tensorrtx wang-xinyu
1.7k
6.7k
mit
102
Implementation of popular deep learning networks with TensorRT network definition API
crnn detr resnet yolov3 yolov4 yolov5 yolov7 yolov8 yolov9 arcface mnasnet tensorrt retinaface squeezenet yolov3-spp inceptionv3 mobilenetv2 mobilenetv3 swin-transformer
Created 2019-11-25
443 commits to master branch, last one 17 days ago
rwightman
gen-efficientnet-pytorch rwightman
213
1.6k
apache-2.0
44
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
108 commits to master branch, last one 2 years ago
Kedreamix
Pytorch-Image-Classification Kedreamix
23
165
unknown
2
用于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 5 months ago
ZJCV
ZCls ZJCV
24
138
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
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
104
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 2 years ago