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

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2023年2月26日 下午1:06
下一篇 2023年2月26日 下午1:07

相关推荐