False positives in faster-rcnn object detection

房东的猫 提交于 2020-02-28 06:59:22

问题


I'm training an object detector using tensorflow and the faster_rcnn_inception_v2_coco model and am experiencing a lot of false positives when classifying on a video.

After some research I've figured out that I need to add negative images to the training process.

How do I add these to tfrecord files? I used the csv to tfrecord file code provided in the tutorial here.

Also it seems that ssd has a hard_example_miner in the config that allows to configure this behaviour but this doesn't seem to be the case for faster rcnn? Is there a way to achieve something similar on faster rcnn?


回答1:


I was facing the same issue with faster RCNN, although you cannot actually use hard_example_miner with the faster RCNN model, you can add some background images, ie. images with no objects (Everything remains the same, except there is not object tag in the xml for that particular picture)

One more thing that actually worked wonders for me was using the imgaug library, you can augment the images and the bounding boxes using the same script. Try and increase the training data by 10 or 15 times, and then I would suggest you to train again to around 150000-200000 steps.

These two steps helped me reduce the number of false positives effectively.



来源:https://stackoverflow.com/questions/54004330/false-positives-in-faster-rcnn-object-detection

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!