Skip to content

MobileNets

Pre-release
Pre-release
Compare
Choose a tag to compare
@ffiirree ffiirree released this 21 Sep 11:32
· 119 commits to master since this release

Accuracy

Model Parms(M) FLOPS(M) Top-1@Paper Top-1 Top-5
MobileNet V1 x1.0 4.232 579 70.6 72.966(73.264@224/232) 91.190
MobileNet V1 x0.75 2.586 333 68.4 69.894 89.520
MobileNet V1 x0.5 1.332 155 63.7 65.590 86.336
MobileNet V1 x0.35 0.766 79 - 59.328 82.068
MobileNet V2 x1.0 3.505 314 72.0 72.154(224/232) 90.736
MobileNet V2 x0.75 2.636 221 - 68.738 88.330
MobileNet V2 x0.5 1.969 104 - 63.926 85.154
MobileNet V2 x0.35 1.677 65.074 - 59.240 81.782

Training

Procedure
learning rate 0.2
batch size 512
lr scheduler cosine
min lr 1e-6
optimizer SGD with momentum 0.9
weight decay 1e-4
epochs 100
warmup epochs 5
lable smoothing 0.1
mixup -
cutmix -
amp True
image processing RandomResizedCrop/RandomHorizontalFlip
data augment -
dropout rate 0.0
drop path rate -
crop size 224
val-resize size 256
val-resize size 224

MobileNet V1 x1.0

Procedure
crop size 192
val-resize size 232
val-resize size 192

V2

Procedure
weight decay 1e-5
data augment normal