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
设置成你自己数据的类别数
更改前代码
parser.add_argument('--data-path', type=str,
default="/data/flower_photos")
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)')
更改后代吗
parser.add_argument('--data-path', type=str,
default="C:/Users/11111/Desktop/pytorch/Test7_shufflenet/flower_photos")
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’是我的文件路径
Original: https://blog.csdn.net/weixin_45868623/article/details/126129044
Author: 咕咕咕不咕
Title: train.py: error: the following arguments are required: –dataset的解决办法
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/614172/
转载文章受原作者版权保护。转载请注明原作者出处!