YuwenXiong / py-R-FCN

R-FCN with joint training and python support
MIT License
1.05k stars 471 forks source link

cls_num in prototxt is not same #33

Open fbi0817 opened 7 years ago

fbi0817 commented 7 years ago

For example, in /models/pascal_voc/ResNet-101/rfcn_end2end/class-aware/test.prototxt line 7054 and line 7079 , the value of cls_num in line 7054 is 21, while in line 7079 is 8, I don't know why, can anyone explain it ? Thank you !

liyi14 commented 7 years ago

Hi @fbi0817. In line 7054, the cls_num is 21 refers to the number of classes in VOC (20 object +1 background). In line 7079, there is a typo in comments. it should be 21(cls_num)4(bbox_reg_channel, dx, dy, dh, dw)7^2(score_map_size^2)