问题
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