train.py: error: the following arguments are required: --dataset的解决办法

train.py: error: the following arguments are required: --dataset解决办法
跑别人项目学习的时候报的错误

对应流程

1.去对应链接下载相关数据文档
2. 在train.py脚本中将--data-path设置成解压后的flower_photos文件夹绝对路径
3. 下载预训练权重，在model.py文件中每个模型都有提供预训练权重的下载地址，根据自己使用的模型下载对应预训练权重
4. 在train.py脚本中将--weights参数设成下载好的预训练权重路径
5. 设置好数据集的路径--data-path以及预训练权重的路径--weights就能使用train.py脚本开始训练了(训练过程中会自动生成class_indices.json文件)
6. 在predict.py脚本中导入和训练脚本中同样的模型，并将model_weight_path设置成训练好的模型权重路径(默认保存在weights文件夹下)
7. 在predict.py脚本中将img_path设置成你自己需要预测的图片绝对路径
8. 设置好权重路径model_weight_path和预测的图片路径img_path就能使用predict.py脚本进行预测了
9. 如果要使用自己的数据集，请按照花分类数据集的文件结构进行摆放(即一个类别对应一个文件夹)，并且将训练以及预测脚本中的num_classes设置成你自己数据的类别数

更改前代码

# 数据集所在根目录
    # https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz
    parser.add_argument('--data-path', type=str,
                        default="/data/flower_photos")

    # shufflenetv2_x1.0 官方权重下载地址
    # https://download.pytorch.org/models/shufflenetv2_x1-5666bf0f80.pth
    parser.add_argument('--weights', type=str, default='./shufflenetv2_x1.pth',
                        help='initial weights path')
    parser.add_argument('--freeze-layers', type=bool, default=False)
    parser.add_argument('--device', default='cuda:0', help='device id (i.e. 0 or 0,1 or cpu)')

更改后代吗

# 数据集所在根目录
# https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz
parser.add_argument('--data-path', type=str,
                    default="C:/Users/11111/Desktop/pytorch/Test7_shufflenet/flower_photos")#/data/flower_photos

# shufflenetv2_x1.0 官方权重下载地址
# https://download.pytorch.org/models/shufflenetv2_x1-5666bf0f80.pth
parser.add_argument('--weights', type=str, default='C:/Users/11111/Desktop/pytorch/Test7_shufflenet/shufflenetv2_x1.pth',
                    help='initial weights path')
parser.add_argument('--freeze-layers', type=bool, default=False)
parser.add_argument('--device', default='cuda:0', help='device id (i.e. 0 or 0,1 or cpu)')

'C:/Users/11111/Desktop/pytorch/Test7_shufflenet/shufflenetv2_x1.pth’是我的文件路径