编辑:发现问题。创建数据集时忘记添加自定义类。业余错误,但留给犯类似错误的其他人。
在成功学习了 NVIDIA 提供的 KITTI 教程后,我开始尝试在自己的数据集上运行 DIGITS。问题是,几乎在开始后,训练和验证的损失覆盖率几乎直接为零,损失为 10e-4。
额外细节:
- 约 900 张训练图像和约 300 张验证图像(注意:我的一位朋友告诉我,他可以使用这个数据集,所以我不认为图像数量是问题)
- 这些图像非常简单,并且包含一个基本背景为普通背景的汽车模型。
- 我必须通过从训练数据中获取图像/标签来创建验证数据
- 我使用的模型是DIGITS 物体检测教程中提到的 DetectNet 模型。
- 我使用了本教程中提到的 GoogLeNet 预训练模型。
我尝试了什么:
- 更改超参数似乎无济于事
- 如前所述,运行模型超过 1 个 epoch 不会改变任何东西(损失覆盖率保持接近于零)