有没有办法在视频中使用Yolov5的两个不同的权重文件？
有没有办法在视频中使用Yolov5的两个不同的权重文件？
提问于 2022-08-05 22:56:40
我有两种经过训练的YoloV5模型用于笔尖和笔帽检测(pen.pt，cap.pt)。
我想用这两种型号制作一个视频。为此，我运行一个命令
! python detect.py --weights cap.pt pen.pt --img 640 --conf 0.50 --source VID_20220727_185703.mp4它运行正常，并分别检测笔和帽，但，它只显示为两个的标签为帽。
有办法解决吗？而不需要重新训练整个数据集。
回答 1
Stack Overflow用户
发布于 2022-08-06 04:56:55
在model ensembling环境下，在YoloV5中实现了使用两种模型对单个数据进行推理的过程。
模型安装教程明确定义：
集成建模是通过使用许多不同的建模算法或使用不同的训练数据集来创建多个不同的模型来预测结果的过程。然后，集合模型聚合每个基本模型的预测，并对未见数据进行一次最终预测。使用集成模型的动机是为了减小预测的泛化误差。
因此，在测试和推理过程中，模型集成可以提高mAP和查全率，但这两种模型应该针对相同的类进行训练。
这一点在issue#1188中得到了澄清。
因此，这里的一个解决办法可能是，使用来自一个推理的输出视频作为第二个模型的推理的输入。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/73257792
复制相关文章
相似问题
腾讯云开发者
