【发布时间】:2020-12-11 20:36:29
【问题描述】:
我正在尝试使用 tensorflow api models ssd mobile net 创建自己的自定义对象检测器,但问题是当模型开始训练时,损失非常高,例如 700-800,并且损失一直在波动,我也看到了同样的情况损失的价值被重复,有人可以向我解释一下。我最初已经将模型训练了 7000 步,但损失并没有减少,然后我又开始训练模型。我正在分享我的新培训的屏幕截图。我已经坚持了将近两天了,非常感谢任何帮助。 我的火车数据集中有 16482 张图像,我的图像大小是 64x64
【问题讨论】:
-
在不提供任何关于您的数据、模型等的信息的情况下,我们只能猜测。你标准化你的输入了吗?您的模型是否足够大以适应数据?有没有可以学习的模式? ...
-
您的损失似乎在波动。尝试降低学习率。否则使用这个 repo 来设置对象 - 检测 - 模型并尝试不同的模型,看看是否会发生同样的事情 - Link
-
@AniketBote 嘿,这是用于自定义对象检测的存储库吗?
-
是的。它会自动下载所有内容并设置路径,创建 tfrecords ,模型模型,创建配置文件,最后输出用于训练和评估模型的命令。
标签: tensorflow object-detection-api mobilenet