张量流对象检测训练模型不起作用

Question

我使用ssd和更快的r-cnn模型训练了用于张量流对象检测的数据集。我的数据集中有220个训练图像和30个测试图像。 我对模型进行了20万步的训练，结果损失在1以下。但是当我在视频上测试训练好的模型时，它几乎可以检测并标记视频中的所有内容。 谁能告诉我为什么会这样吗？ 谢谢

Answer 1

您正在使用的类的数量仅为一种，并且您使用属于同一类的图像训练了模型并对其进行了测试。

因此，问题在于模型是偏斜的（对所有图像预测相同），无论在哪个图像上进行测试，您都将获得相同的输出。

解：

用几乎相等数量的负像训练您的模型。

例如：包含要识别的对象的220张图像（将其标记为1）和不包含对象的另外近220张图像（将其标记为0）

使用F1分数来检查您的准确性，因为它可以帮助您了解数据集是否倾斜。

选中此选项可了解不同类型的准确性度量。 参加本课程以了解有关CNN的更多信息。