我已经基于张量流模型动物园https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md
中的ssd_mobilenet_v2_quantized_coco
调整了对象检测模型。
我的微调模型始终预测4个对象,而与输入的图像无关。
是什么导致模型学习始终准确预测4个对象?
我已验证我的训练数据在图像中包含大量对象。我尝试使用基本模型zip内的默认配置文件,以及在以下位置签入github的那个文件:https://github.com/tensorflow/models/tree/master/research/object_detection/samples/configs。两者产生相同的结果。
我的管道配置文件基于https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md或https://github.com/tensorflow/models/tree/master/research/object_detection/samples/configs中ssd_mobilenet_v2_quantized_coco
模型的默认值。除了填写相关的本地路径并使用我的自定义标签图,没有其他修改。
我希望我的模型应该能够预测0-100个对象。相反,它总是预测为4。
示例输出: (标签):(信心)
图片1(全部错误):
STATIONARY_CAR: 0.0820312
PERSON: 0.0820312
PERSON: 0.0585938
STATIONARY_CAR: 0.0585938
图片2(汽车正确,位置错误):
STATIONARY_CAR: 0.992188
PERSON: 0.0898438
PERSON: 0.0585938
PERSON: 0.0585938
图片3(全部正确):
STATIONARY_CAR: 0.910156
STATIONARY_CAR: 0.910156
STATIONARY_CAR: 0.867188
STATIONARY_CAR: 0.613281