使用SSD Mobilenet(Tensorflow API)在1080p中进行对象检测

时间:2019-10-14 18:25:09

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

大家好,

我的目标是在1920x1080大小的图像上检测人和汽车(白天和黑夜),为此,我使用了tensorflow API,SSD mobilenet模型,并注释了1000张图像(用于训练的900张图片,用于100张的评估)来自7个不同的相机我以960x540的图像尺寸启动了培训。我的模型无法收敛。我不知道该怎么办,我应该为昼夜物体设置不同的类吗?

在使用tensorflow API进行人脸检测的教程中,他们使用仅包含人脸的图像的数据集,然后在复杂场景上使用模型。知道类似SSD的模型也会学习负面示例,这是一个好主意吗?

谢谢

(来源:https://blog.usejournal.com/face-detection-for-cctv-surveillance-6b8851ca3751

1 个答案:

答案 0 :(得分:0)

“不收敛”是什么意思?您是指火车/验证损失吗?
在这种情况下,我想到的第一件事就是降低学习率(我有类似的问题)。 您可以通过修改配置文件来做到这一点,在“ train_config ”部分中,您将找到值“ initial_learning_rate ”。
尝试将其设置为一个较低的值(例如,降低一个数量级),看看是否有帮助。