简述

yolov5中推理前的图像预处理过程

内容

1.resize

  • 中心裁剪方式:从图像size的最小边中心进行裁剪——分类模型

yolov5 图像预处理_yolov5

  • 短边填充方式(letterbox):短边上去填充0像素,使得图像扩充为网络输入尺寸——检测模型和分割模型

yolov5 图像预处理_yolov5_02

2.blob

  • BGR转为RGB

yolov5 图像预处理_yolov5_03

  • 归一化

yolov5 图像预处理_yolov5_04

  • 分类模型比检测模型、分割模型多了个规范化操作(减去均值再除以方差)

yolov5 图像预处理_yolov5_05