shaoshengsong / MobileNetV3-SSD

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

V3是否有在imagenet预训练过?没有预训练的效果很差 #24

Open johsnows opened 4 years ago

johsnows commented 4 years ago

Average Precision Per-class: aeroplane: 0.5286851217064079 bicycle: 0.5684498038841389 bird: 0.2951959818198857 boat: 0.3374119115412078 bottle: 0.11274539710602145 bus: 0.5840837968782298 car: 0.5631029936677479 cat: 0.5605063414099851 chair: 0.24291365433556902 cow: 0.2908948526148929 diningtable: 0.4292561103683377 dog: 0.47845551032801703 horse: 0.6137629939581943 motorbike: 0.6033233235496501 person: 0.5106418290063527 pottedplant: 0.13170114489968404 sheep: 0.3551660800668141 sofa: 0.4682129143642543 train: 0.6344977720902926 tvmonitor: 0.4312501217962447

Average Precision Across All Classes:0.4370128827695964

shaoshengsong commented 4 years ago

你试试第二个版本(https://github.com/shaoshengsong/MobileNetV3-SSD-Compact-Version),损失能降到1.9,1.8左右