From 2060e433c02d80c117635acc3267ab6cb3515824 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 11 Jun 2019 08:54:54 -0700 Subject: [PATCH] Add native PyTorch weights for SE-MnasNet aka MnasNet-A1 --- README.md | 1 + models/gen_efficientnet.py | 14 +++++++++----- 2 files changed, 10 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index c2313cba..bfb030f3 100644 --- a/README.md +++ b/README.md @@ -69,6 +69,7 @@ I've leveraged the training scripts in this repository to train a few of the mod | seresnext26_32x4d | 77.104 (22.896) | 93.316 (6.684) | 16.8M | bicubic | | efficientnet_b0 | 76.912 (23.088) | 93.210 (6.790) | 5.29M | bicubic | | mobilenetv3_100 | 75.634 (24.366) | 92.708 (7.292) | 5.5M | bicubic | +| semnasnet_100 | 75.448 (24.552) | 92.604 (7.396) | 3.89M | bicubic | | fbnetc_100 | 75.124 (24.876) | 92.386 (7.614) | 5.6M | bilinear | | resnet34 | 75.110 (24.890) | 92.284 (7.716) | 22M | bilinear | | seresnet34 | 74.808 (25.192) | 92.124 (7.876) | 22M | bilinear | diff --git a/models/gen_efficientnet.py b/models/gen_efficientnet.py index aef2c1b9..ecd4f65a 100644 --- a/models/gen_efficientnet.py +++ b/models/gen_efficientnet.py @@ -51,14 +51,18 @@ default_cfgs = { 'mnasnet_050': _cfg(url=''), 'mnasnet_075': _cfg(url=''), 'mnasnet_100': _cfg(url=''), - 'tflite_mnasnet_100': _cfg(url='https://www.dropbox.com/s/q55ir3tx8mpeyol/tflite_mnasnet_100-31639cdc.pth?dl=1', - interpolation='bicubic'), + 'tflite_mnasnet_100': _cfg( + url='https://www.dropbox.com/s/q55ir3tx8mpeyol/tflite_mnasnet_100-31639cdc.pth?dl=1', + interpolation='bicubic'), 'mnasnet_140': _cfg(url=''), 'semnasnet_050': _cfg(url=''), 'semnasnet_075': _cfg(url=''), - 'semnasnet_100': _cfg(url=''), - 'tflite_semnasnet_100': _cfg(url='https://www.dropbox.com/s/yiori47sr9dydev/tflite_semnasnet_100-7c780429.pth?dl=1', - interpolation='bicubic'), + 'semnasnet_100': _cfg( + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_a1-d9418771.pth', + interpolation='bicubic'), + 'tflite_semnasnet_100': _cfg( + url='https://www.dropbox.com/s/yiori47sr9dydev/tflite_semnasnet_100-7c780429.pth?dl=1', + interpolation='bicubic'), 'semnasnet_140': _cfg(url=''), 'mnasnet_small': _cfg(url=''), 'mobilenetv1_100': _cfg(url=''),