yolov5+deepstream中，训练图像大小、img-size、usb摄像头的图像尺寸关系？_yolo能训练1920*1080的图片吗

作者：羊村懒王 | 2024-04-06 18:13:50

踩

yolo能训练1920*1080的图片吗

（1）在yolov5的train.py，将img-size参数设置为[1920,1920]，训练得到权重pt文件，并转换为wts文件；训练图片尺寸是19201080
（2）然后把tensorrt的yololayer.h的INPUT_H和INPUT_W设置为1920，将wts转换为engine；（3）最后在Deepstream5.1中，将source1_usb_dec_infer_yolov5.txt的camera-width设置为1920，将camera-height设置为1080（摄像头支持1920*1080）。再运行deepstream-app -c source1_usb_dec_infer_yolov5.txt，出现以下问题。在这里插入图片描述

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/羊村懒王/article/detail/373496