为什么我的模型会学习总是预测4个对象?

时间:2019-07-05 22:18:33

标签: tensorflow deep-learning object-detection object-detection-api

我已经基于张量流模型动物园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.mdhttps://github.com/tensorflow/models/tree/master/research/object_detection/samples/configsssd_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

0 个答案:

没有答案