search

facebookresearch / detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
https://detectron2.readthedocs.io/en/latest/
Apache License 2.0
29.92k stars 7.4k forks source link

how to train R-FCN on custom dataset? #360

Open ravising-h opened 4 years ago

ravising-h commented 4 years ago

❓ Questions and Help

General questions about detectron2. Hi! I am new to Object detection. Can Someone guide me with this. How to train R-FCN using Detectron2.

ppwwyyxx commented 4 years ago

It's not included in detectron2 and you'll need to implement it yourself

TemitopeOladokun commented 4 years ago

I tried to register my custom coco dataset but I got this error message

"Category ids in annotations are not in [1, #categories]! We'll apply a mapping for you." More so, I decided to print the content of the folder, it cannot be read.

Please what am I doing wrong?

j-adamczyk commented 4 years ago

@TemitopeOladokun this is ok, it's just a warning message. It means that your category IDs were not from 1 to number of categories, but something else, most likely you indexed from 0. Detectron2 will automatically remap your category IDs to match this. Beware though that you have to create the same mapping for your classification examples, so you may want to do it by hand anyway before using NN.