0


pytorch yolov5的输入图像尺寸为指定尺寸

yolov5支持两种训练方式:
假如指定输入img-size为640

square (w==h)
如 输入为 [b, c, 640, 640], 可以使用mosic数据增强方式增强图像

rect(scale):
如 输入为 [b, c, 640, 512], 其中512为短边放缩以后的尺寸(补充到32的倍数)
但是不支持mosic数据增强方式

但是有的时候在实际项目使用中,可能会涉及到需要同时指定输入图像的长和宽,因此对yolov5代码做一些修改以适应于项目需求。

修改后的代码如下:

mirrors / shanglianlm0525 / yolov5_specific_size · GitCode
GitHub - shanglianlm0525/yolov5_specific_size: yolov5 project that can specific size of the input image, both width and height.

(31条消息) 修改yolov5的输入图像尺寸为指定尺寸_mingo_敏的博客-CSDN博客_yolov5输入图片尺寸


本文转载自: https://blog.csdn.net/Vertira/article/details/127434975
版权归原作者 Vertira 所有, 如有侵权,请联系我们删除。

“pytorch yolov5的输入图像尺寸为指定尺寸”的评论:

还没有评论