search

mrlooi / rotated_maskrcnn

Rotated Mask R-CNN: From Bounding Boxes to Rotated Bounding Boxes
MIT License
347 stars 62 forks source link

how to set the number of class? #17

Closed ustczhouyu closed 4 years ago

ustczhouyu commented 4 years ago

❓ Questions and Help

Thank you for sharing the code. I would like to know, if there are 10 classes in my dataset, which parameters under which files do I need to change to 11? Looking forward to your reply, thank you @mrlooi

mrlooi commented 4 years ago

Hey, if you look at maskrcnn_benchmark/config/defaults.py, there's a field called _C.MODEL.ROI_BOX_HEAD.NUM_CLASSES. It's 81 by default (num COCO classes + 1 background class). Add this to your .yaml config, and change the value to your num classes + 1 (for background class)

ustczhouyu commented 4 years ago

Thanks for your reply. I have set it up. Thank you again

zhou19920226

邮箱：zhou19920226@126.com |

Signature is customized by Netease Mail Master

On 01/19/2020 11:16, mrlooi wrote:

Hey, if you look at maskrcnn_benchmark/config/defaults.py, there's a field called _C.MODEL.ROI_BOX_HEAD.NUM_CLASSES. It's 81 by default (num COCO classes + 1 background class). Add this to your .yaml config, and change the value to your num classes + 1 (for background class)

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

mrlooi commented 4 years ago

Glad I could help, closing this issue

ustczhouyu commented 4 years ago

Hello, I want to know how to convert the test results of ICDAR into the format required by ICDAR15 for evaluation. Please help me, thank you！

zhou19920226

邮箱：zhou19920226@126.com |

Signature is customized by Netease Mail Master

On 01/19/2020 11:16, mrlooi wrote:

Hey, if you look at maskrcnn_benchmark/config/defaults.py, there's a field called _C.MODEL.ROI_BOX_HEAD.NUM_CLASSES. It's 81 by default (num COCO classes + 1 background class). Add this to your .yaml config, and change the value to your num classes + 1 (for background class)

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

mrlooi commented 4 years ago

I haven't tried icdar evaluation, they should have plenty of docs on that. Good luck!

ustczhouyu commented 4 years ago

Ok，thank you,I will try

zhou19920226

邮箱：zhou19920226@126.com |

Signature is customized by Netease Mail Master

On 01/23/2020 10:33, mrlooi wrote:

I haven't tried icdar evaluation, they should have plenty of docs on that. Good luck!

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.