yolov7 model 检测两类目标人脸和抽烟

Question

我用yolov7 model 检测两类目标人脸和抽烟，通过train训练完成得到best.pt模型权重文件，通过detect.py检测图片，只出现一个标记框，另外一个没有标记或者两个标记成一样的，这种问题如何解决？

Answer 1

回答不易，求求您采纳点赞哦

这种情况通常是因为模型训练不够充分或训练数据不够完整，也有可能是检测算法的参数需要调整，你可以尝试以下几个解决方案:

• 增加训练数据量并重新训练模型；
• 对训练数据进行数据增强，以提高模型泛化能力；
• 调整检测算法的阈值，以减小标记框的数量；
• 检查检测算法是否适用于你的数据，并考虑替换其他算法。