目标检测数据预处理--尺寸变换

目录

1. 比例缩放

2. 使用letterbox

---

目标检测训练中，我们的数据集尺寸大部分时侯都是不符合网络输入的，需要对尺寸进行修改，下面我介绍两种常用尺寸变换方法：

1. 比例缩放

这种方法就是简单的对图片尺寸进行比例缩放，一般使用cv2.resize()对图片进行缩放，然后计算长宽缩放比例，再通过比例来缩放标注的目标框尺寸。具体代码如下：

def read_and_resize_picture(img_path, img_boxes):
    """
    :function:读取图片，并缩放图片尺寸到指定尺寸，同时修改目标框尺寸
    :param img_path: 单张图片路径
    :param img_boxes: 对应图片目标框[[xmin,ymin,xmax,ymax],...,[xmin,ymin,xmax,ymax]]