Skip to content
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

Small documentation issues for train&test ResNet-50 (without OHEM) #1

Closed
ngaloppo opened this issue Sep 12, 2016 · 4 comments
Closed

Comments

@ngaloppo
Copy link

Thanks for making this code available. I am trying to train & test ResNet-50 (without OHEM). Here are the issues I have found in the docs:

  • Have to run make in $FRCN_ROOT/lib/ folder for cython_bbox compilation
  • Before running the training script, make sure your add $CAFFE_ROOT/python to your PYTHONPATH environment variable
  • The training script takes an extra argument for the dataset (pascal_voc or coco). This is not documented in the README, but it is documented in the script itself.
  • Despite what is stated in the README, test.agonistic and train.agonistic are set to True by default by the cfgs/rfcn_end2end.yml
@YuwenXiong
Copy link
Owner

YuwenXiong commented Sep 13, 2016

Hi @ngaloppo , thanks for your attention, I indeed made something unclear. And I followed your advice, modified the document. In fact, AGONISTIC is set to False by default, you could check lib/fast_rcnn/config.py, but I provide a configuration file for you, you could just use this file without any modification.

Can you run this code now? I'm looking forward to seeing someone who run this code successfully too. And my result shows that the result of end-to-end training is not as good as 4-step training. It seems that when you use joint training, poor rois that rpn produces at the very beginning will "confuse" OHEM, but I'm not sure, it needs more time to figure it out.

@ngaloppo
Copy link
Author

Hello @orpine. Thanks again for providing this code. Regarding the AGONISTIC setting, I think my point is that if the user doesn't do anything special, the script will automatically use the experiments/cfgs/rfcn_end2end.yml file, and so AGONISTIC is automatically set to True. Just a minor point though...

I was able to train & test. I'm getting a mean-AP of 0.7405 for VOC2007 test data on the VOC2007+2012 training data. I didn't apply OHEM during training, so I can't directly compare to the results you have posted in the README, but I believe that this is a great result. Would you agree?

@xiaoxiongli
Copy link

@ngaloppo yes, I agree^_^, but unfortunately, I am not familiar with python...

@YuwenXiong
Copy link
Owner

@ngaloppo Yes, I think that is a reasonable result, and I found that sometimes 110k achieves better results than 120k, I don't know why, maybe because I use 12000/2000 rois instead of 6000/300 rois in the original code, I'm trying to figure it out, and you can check the 110k result by run test_net.py individually :).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants