mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

FBNetV3 timm trained weights added for b/d/g variants. Update version to 0.5.2 for pypi release.

Browse Source
This commit is contained in:
Ross Wightman 2022-01-07 12:05:08 -08:00
parent 02ae11e526
commit 1406cddc2e
3 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -24,9 +24,11 @@ I'm fortunate to be able to dedicate significant time and money of my own suppor
## What's New
### Jan 6, 2022
* Tried training a few smaller models, two are good so far, more on the way...
* Version 0.5.2 w/ release to be pushed to pypi. It's been a while since last pypi update and riskier changes will be merged to main branch soon....
* Tried training a few small / mobile optimized models, a few are good so far, more on the way...
* `mnasnet_small` - 65.6 top-1
* `lcnet_100` - 72.1 top-1
* `fbnetv3_b/d/g` - 79.1 / 79.7 / 82.0
* TinyNet models added by [rsomani95](https://github.com/rsomani95)
* LCNet added via MobileNetV3 architecture

12
timm/models/mobilenetv3.py
View File

@ -75,9 +75,15 @@ default_cfgs = {
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_mobilenetv3_small_minimal_100-922a7843.pth',
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD),
'fbnetv3_b': _cfg(),
'fbnetv3_d': _cfg(),
'fbnetv3_g': _cfg(),
'fbnetv3_b': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_b_224-ead5d2a1.pth',
test_input_size=(3, 256, 256), crop_pct=0.95, pool_size=(8, 8)),
'fbnetv3_d': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_d_224-c98bce42.pth',
test_input_size=(3, 256, 256), crop_pct=0.95, pool_size=(8, 8)),
'fbnetv3_g': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_g_240-0b1df83b.pth',
test_input_size=(3, 288, 288), crop_pct=0.95, pool_size=(9, 9)),
"lcnet_035": _cfg(),
"lcnet_050": _cfg(),

2
timm/version.py
View File

@ -1 +1 @@
__version__ = '0.5.0'
__version__ = '0.5.2'