Open askerlee opened 7 years ago
@askerlee @Eniac-Xie Could you please release your "ResNet101_BN_SCALE_Merged_OHEM" model files included test.prototxt,my download files not contain it,but i have no time to write it because of an emergency.Thank you very much!!!
Just copy the test.prototxt from the ResNet101_BN_SCALE_Merged folder. They are the same (hard example mining only happens in training, so the test model is the same).
Ok,thank you!!!Now i got an error when run ./experiments/scripts/faster_rcnn_end2end.sh 0 ResNet-50 pascal_voc,like this: i did not change the original code about resnet-50 with ohem,but only replace the "num_classes" and "num_output",i can not solve it,could you help me?
change cls_prob[i,label]
to cls_prob[i,int(label)]
in lib/roi_data_layer/layer.py:242.
Hi, I have encountered the same problem. Have you solved it?I think it may be the problem of the code. When I used the OHEM code modified by myself to train the author's prototxt file, the loss was not 0, but it was difficult to converge (which did not exist on VGG16).
Hi, I'm using ResNet101_BN_SCALE_Merged_OHEM on my own dataset. Some of the output losses (loss_bbox and loss_cls) are always 0.
Update: seems there are something wrong with OHEM. When I turn off OHEM everything is normal.
The output with OHEM turned off: