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输入图片尺寸