训练方法咨询

[zhjygit]

师傅你好，按照项目里面的训练方法，我下载了预训练权重，能识别的类型是：aeroplane bicycle bird boat bottle bus car cat chair cow diningtable dog horse motorbike person pottedplant sheep sofa train tvmonitor 。 我像识别的目标是工业安全帽helmet，将其添加到最后一个。然后准备了1500张左右的安全帽图片，并对xml文件、jpg等进行了手动整理，voc_annotation.py运行正常。 接着运行train.py： 简单做了训练测试，gpu是16G显存，占用率30%左右，但迭代速率很低只有1.5左右。训练20分钟左右后，将新的权重加载并预测。 预测结果非常郁闷的是，安全帽识别正常，但半身人像也被识别为了安全帽： 我的数据集也手动清理过无关的照片，整体没有单独的半身像： 非常郁闷，到底是哪里出问题了呢