shaoshengsong / MobileNetV3-SSD

MobileNetV3-SSD for object detection and implementation in PyTorch
269 stars 88 forks source link

Mobilenetv3_py is outdated #23

Open CoskunGorkem opened 4 years ago

CoskunGorkem commented 4 years ago

When I compare the implementation of yours and official paper, I found some differences probably due to the paper update in last November.

Changes are in Mobilenet_v3.py: line 159
[112, 160, 5, 1, "HS", True, 672] needs to be [112, 160, 5, 2, "HS", True, 672]

line 159 [160, 160, 5, 2, "HS", True, 672] needs to be [160, 160, 5, 1, "HS", True, 960]

If I am not mistaken, Line 246 out_conv2_out = _make_divisible(1280 * multiplier) needs to be

out_conv2_out = _make_divisible(1024 * multiplier), since this is MobilenetV3_Small. Mobilenetv3_Large uses 1280.

out_conv2_out = _make_divisible(1280 * multiplier)