Tensorflow-Custom对象检测Api头盔检测

Question

我正在尝试使用Tensorflow Object Detection Api检测未戴头盔的人。我正在使用3000张图像（不戴头盔的人1000张，戴头盔的人1000张，人1000张）。我使用SSD_mobilenet_v1_pets.config进行训练。 请找到PBTXT文件：

item {
    id: 1
    name: 'withouthelmet'
}
item {
    id: 2
    name: 'withhelmet'
}
item {
    id: 3
    name: 'person'
}

在SSD_mobilenet_v1_pets.config文件中，（id：1是供人使用，但我已将id：1设置为'withouthelmet'，id：2为'withhelmet'，id：3为'person'）。

训练模型后，它会将每个人都检测为“没有头盔”，将每辆车都检测为“人”。

您能告诉我问题出在哪里吗？