Tensorflow对象检测API-测试问题

Question

我有2000张图像（头盔为1000张，未戴头盔的人为1000张）。经过23k的训练步骤后，我使用custom_inference.py文件（I use this python file测试了视频。它将整个帧检测为头盔。这是我得到的输出

我的pbtxt文件是

item {
  id: 1
  name: 'Helmet'
}
item {
  id: 2
  name: 'withouthelmet'
}

这是我用于训练的示例图像

请帮我解决这个问题。