PYTHON:将图像缩放至固定尺寸，并生成相应的路径列表

import cv2 as cv
import os


def resize_img(src_dir, dst_dir, imgList416, img_size):
    '''
    :param src_dir: train.txt dir
    :param dst_dir: 416*416 img save dir
    :param imgList416: 416*416 img list txt file
    :param img_size: img scale
    :return:
    '''
    w = img_size[0]
    h = img_size[1]
    file = open(src_dir, 'r')
    imgList416File = open(imgList416, 'w')
    imgDirLists = file.readlines()
    for imgDir in imgDirLists:
        imgDirNew = imgDir.replace('person_pic', 'rknn416')
        imgList416File.write(imgDirNew)
        imgDir=imgDir.rstrip('\n')
        img = cv.imread(imgDir)
        imgRes = cv.resize(img, (w, h), interpolation=cv.INTER_LINEAR)

        imgDirSplit = imgDir.split('/')

        if os.path.exists(dst_dir):
            imgSaveDir = dst_dir + '/' + imgDirSplit[-1]
            cv.imwrite(imgSaveDir, imgRes)
        else:
            os.mkdir(dst_dir)
            imgSaveDir = dst_dir + '/' + imgDirSplit[-1]
            cv.imwrite(imgSaveDir, imgRes)
    file.close()
    imgList416File.close()


if __name__ == '__main__':
    src_dir = '/home/.../extra/personDataset/train.txt'
    dst_dir = '/home/.../extra/personDataset/rknn416'
    imgList416 = '/home/.../extra/personDataset/imgList416.txt'
    img_size = [416, 416]
    resize_img(src_dir, dst_dir, imgList416, img_size)