首页 > 解决方案 > 为什么我在训练tensorflow对象检测ssd mobilenet模型时训练损失很高

问题描述

我正在尝试使用 tensorflow api models ssd mobile net 创建自己的自定义对象检测器,但问题是当模型开始训练时,损失非常高,如 700-800,并且损失不断波动,我看到相同的损失值重复,有人可以向我解释一下。我最初已经将模型训练了 7000 步,但损失并没有减少,然后我又开始训练模型。我正在分享我的新培训的屏幕截图。我已经坚持了将近两天了,非常感谢任何帮助。我的火车数据集中有 16482 张图像,我的图像大小是 64x64

失利

标签: tensorflowobject-detection-apimobilenet

解决方案


请检查图像格式所有图像应为相同格式(JPG 或 JPEG),删除 PNG 图像。并尝试训练


推荐阅读