search

houqb / CoordAttention

Code for our CVPR2021 paper coordinate attention
MIT License
998 stars 121 forks source link

Add coordattention in MobileNetV2? #18

Open cc19860606 opened 3 years ago

cc19860606 commented 3 years ago

97EE$IWN9Z}SO~)DBU0}L`B I modify the code of MobileNetV2 by adding CoordAttention at line 101, is it right? @Andrew-Qibin

cc19860606 commented 3 years ago

Anyone reproduct the performance of CoordAttention with MobileNetV2 on VOC2007?

houqb commented 3 years ago

Have you pretrained the model on ImageNet?

cc19860606 commented 3 years ago

I firstly test MobileNetV2,the mAP can achieve 71.7. Then, I try to add CoordAttention as shown above, meanwhile the pretrained model of 'mobilenet_v2-b0353104.pth' is not use. The mAP only can achieve 64. @Andrew-Qibin

cc19860606 commented 3 years ago

ssdlite-pytorch-mobilenext with CoordAttention need to be trained with ImageNet, then test on VOC?

houqb commented 3 years ago

Exactly. Otherwise, it does not work. Recently, I will release the pretrained models.

cc19860606 commented 3 years ago

I look forward for it, thx.

cc19860606 commented 3 years ago

The ImageNet dataset used for training is ILSVRC2012 or others? And the hyperparameter and code of data augmentation for training? @Andrew-Qibin