pytorch-faster-rcnn
pytorch-faster-rcnn copied to clipboard
How to use self trained resnet in faster rcnn
I am trying to use a self trained resnet101 in faster rcnn. I think that i can get more accurate inferences and results by using one. The issue that I have is when training starts, it throws a Pdb error. This error occurs inside _sample_rois() function located in proposal_target_layer.py. In this function after calculating fg_inds and bg_inds, there are several conditions as shown in below screenshot.
And according to the conditions the control of the program goes into else(i.e. Pdb block) iff fg_inds and bg_inds both are zero simultaneously.
Just for an example,
In below screenshot I have used resnet50 from the link provided in GitHub repo, and after some iterations Pdb error occurs. (I have printed fg_inds and bg_inds for preview)
And if I use a self trained resnet(on a custom dataset), then also it throws a Pdb error, even before the iterations start.
So, do I need to train the resnet in caffe(on custom dataset), then convert to pth and then use in faster rcnn? Will that solve this Pdb error? (Right now I have trained Resnet in pytorch.)
Or is it that self trained resnets are not supported as of now?
And pretrained res50(the one provided in this repo) doesnt work for me either.
Please get me with a solution as soon as possible. It is really important.
Thanks in advance
I am posting first time on GitHub, so I'm sorry if there are any mistakes.
Hello, I have the same problem. Have you solved it?