search

mjq11302010044 / RRPN_pytorch

RRPN in pytorch 1.0 ----- Arbitrary-Oriented Scene Text Detection via Rotation Proposals
MIT License
262 stars 56 forks source link

complete tool/test_net.py for RotationDataset #58

Open markson14 opened 4 years ago

markson14 commented 4 years ago

Since your repo is not able to use inference code for RotationDataset, I have tried to write my own code by referencing Pascal_VOC inference code. I've tested it locally and it works fine to me. Please notice me if anything went wrong.

markson14 commented 4 years ago

I find out the output mAP is a bit wrong due to the following issues so I fix the code.

  1. in transform/build.py, the train and test pipeline are using the same transform which is suppose to be split up
  2. when inferencing RRPN model, the final output boxes' width and height should be divived by cfg.MODEL.RRPN.GT_BOX_MARGIN(default: 1.4)

After fixing the issues, the test_net.py is able to show the correct output.