-
Notifications
You must be signed in to change notification settings - Fork 70
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
VGG 16 (Teacher) mAP is lower than the reported #11
Comments
Hi, have tested the student model? |
Got the issue ! I was using pytorch version 1.0 which was giving lower accuracy. But using pytorch 0.4.0 gave accuracy as 69.96. Still, it is a bit lower than the reported 70.1 The accuracy obtained on Tecaher model are as follows: AP for aeroplane = 0.7351 |
Sorry maybe the model is not exactly the 70.1 AP model |
Using pytorch 1.2, I get the same result. However, the student model, using VGG11, get the 0.68 mAP as reported. |
Sorry I cannot figure out where could possibly be wrong, I recommend you to directly train teacher and student model, vgg16-faster-rcnn with ~70mAP is easy to reproduce. |
@twangnh @yuanli2333 :
lib/model/faster_rcnn/vgg11.py |
Because in this layer, the features of student and teacher model have the same channel (512), but with different spatial size. |
@yuanli2333 : |
Right, the spatial size is the same for VGG16 and VGG11. But we still need one learnable layer to project/transform the teacher feature, the channels also are not exactly one-to-one mapping between teacher and student. |
Great, appreciate your reply. |
We used the the pretrained weights of VGG16 whose link is drive link is shared and tested its accuracy using the test_net.py script but the mAP obtained is 66.06 while the reported mAP on teacher model is 70.1
The detailed output is mentioned below:
AP for aeroplane = 0.6706
AP for bicycle = 0.7505
AP for bird = 0.6550
AP for boat = 0.4441
AP for bottle = 0.4730
AP for bus = 0.7368
AP for car = 0.7654
AP for cat = 0.8224
AP for chair = 0.4576
AP for cow = 0.6890
AP for diningtable = 0.6088
AP for dog = 0.7978
AP for horse = 0.8124
AP for motorbike = 0.7429
AP for person = 0.7343
AP for pottedplant = 0.3794
AP for sheep = 0.5846
AP for sofa = 0.6520
AP for train = 0.7236
AP for tvmonitor = 0.7121
Mean AP = 0.6606
Please let us know if we are missing something. How can we exactly reproduce the exact mAP of the teacher model
The text was updated successfully, but these errors were encountered: