How to use “ignore” class with tensorflow object detection API?

☆樱花仙子☆ 提交于 2020-01-25 05:19:26

问题


I have trained tensorflow object detection model (for num_steps:50000) using SSD (mobilenet-v1) on custom dataset. I got mAP@.50IOU ~0.98 and loss ~1.17. The dataset consist of uno playing card images (skip, reverse, and draw four). On all these cards, model performs pretty well as I have trained model only on these 3 card (around 278 images with 829 bounding boxes (25% bounding box used for testing i.e. validation) collected using mobile phone).

However, I haven’t trained model on any other card but still it detects other cards (inference using webcam).

How can I fix this? Should I also collect other class images (anything other than skip, reverse and draw four cards) and ignore this class in operation? So that model sees this class i.e. Label: Other images during training and doesn’t put any label during inference.

I am not sure how to inform tensorflow object detection API that it should ignore images from Other class.

Can anyone please provide pointer?

Please share your views!

来源:https://stackoverflow.com/questions/59769698/how-to-use-ignore-class-with-tensorflow-object-detection-api

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