YOLO V5 测试图像时 img_size 的设置问题

最近项目用到了 yolo v5。初始图像是 1440×1080 大小的，在训练时显示 “cuda out of memory”，故保持原始长宽比，将图像缩小成 720×540 大小进行训练。问题出在检测过程。

在原始 detect.py 中，有一段定义代码：

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--weights', nargs='+', type=str, default='', help='model.pt path(s)')
    parser.add_argument('--source', type=str, default='', help='source'