问题描述:yolov5训练模型跑出来之后立即进行了图片预测,但是得出的结果是没有对图片进行任何处理。但是摄像头可以正确预测。由此可以得出模型应该是没有问题的,很有可能是图片尺寸问题,于是我便将图片截图为480×640的大小,img-size标签是640。如图:
此时就可以的出预测结果,虽然准确度不高,但是这是与训练集的大小和清晰度有关,就是另一个问题了。得出的结果如下图:
此时已经找到问题的方向了。但是每次都这样截图就会很麻烦,于是就使用了cv2模块中的resize方法对图片进行缩放。同样将图片缩放为480×640。但是此时问题出现了,预测模块仅仅预测到了一处目标,如图:
对图片进行缩放后模型仅仅能预测到一处目标,这与手动截图得出的结果相差甚远。目前正处于学习阶段,对这一问题无法正确解决,望大佬指教。
使用yolov5目标检测算法中进行图片预测遇到的问题
最新推荐文章于 2024-06-18 17:09:16 发布