pytorch YOLOv5未检测到自定义数据

9q78igpj  于 2023-10-20  发布在  其他
关注(0)|答案(1)|浏览(181)

我在使用YOLOv 5时遇到了一个问题,即使在训练之后,它也无法检测到自定义数据中的对象。我试着在Python 3.9和Python 3.10上运行它,结果都是一样的。
%cd yolov5/!python train.py--img 416 --batch 16 --epochs 100 --data /content/yolov5/idk-1/data.yaml --weights yolov5s.pt--cache !python detect.py--weights /content/yolov5/runs/train/exp/weights/best.pt--img 416 --source/content/yolov5/idk-1/train/images/1_jpeg.rf.6095f91252eee827890508b0eb7db4f5.jpg--conf 0.25
然而,该模型没有检测到它所训练的图像中的任何对象。我希望它能探测到训练过的物体。
注意:我使用谷歌colab
其他信息:
我已经检查了我的自定义数据是否正确标记。我验证了数据和权重的路径是正确的。我尝试调整置信度阈值(--conf),但没有成功。任何见解或建议,什么可能导致这个问题将不胜感激。谢谢你,谢谢!
我尝试调整置信度阈值(--conf),但没有成功。任何见解或建议,什么可能导致这个问题将不胜感激。谢谢你,谢谢!

nhhxz33t

nhhxz33t1#

1.在训练数据上测试模型:检查它是否正常工作。如果它确实得到了检测,那么通过这样做,您可以验证您的数据是否已被正确注解,这将导致第二点。
1.验证您的图像已正确注解。如果这也是合适的,那么问题可能与数据集本身的困难有关。
1.可能是您处理的图像具有非常小的尺寸,低分辨率或两者兼而有之的对象。然后,你应该应用一些预处理技术来处理这个问题,比如超分辨率。例如,这是 * 遥感 * 的一个常见问题。您可能需要为此寻找SuperYOLO,或者使用一些生成技术来提高图像分辨率。

相关问题