一、配置参数 argparse.ArgumentParser()
模型路径,输入路径,输出路径,图像大小,处理器选择(cpu gpu),等
二、进入detect()过程
1、torch.device()
2、模型加载 torch.load(模型的路径)
3、加载数据(从某个文件夹中找到多幅图像的路径)
对于每一幅图像执行以下4、5、6操作
4、pred = model(img),其中 img为tensor
5、在inference的结果上执行non_max_suppression(pred)
6、检测结果的保存和显示