yolov5+deepstream中,训练图像大小、img-size、usb摄像头的图像尺寸关系?
(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*10
原创
2021-05-20 15:41:32 ·
7324 阅读 ·
6 评论