Workaround a crash when inference on a picture yielded no predictions.

mrlooi / rotated_maskrcnn

Rotated Mask R-CNN: From Bounding Boxes to Rotated Bounding Boxes

MIT License

347 stars 62 forks source link

Workaround a crash when inference on a picture yielded no predictions. #1

Closed Agade09 closed 4 years ago

Agade09 commented 4 years ago

Hi,

I had an issue when running inference with a custom trained model. Some images yielded no predictions and caused a crash in my_tools/predictor.py. I don't know if it's the cleanest but here is a PR to fix that.

Thanks for your work.

PS: Perhaps you should consider allowing "Issues" in your admin panel?

mrlooi commented 4 years ago

Hey, thanks for the pull request - I've updated the code in the latest commit that includes your changes.

Also, thanks for letting me know about the "issues" panel, just added