源码下载链接:ultralytics/yolov5: YOLOv5 in PyTorch > ONNX > CoreML > TFLite (github.com)
需要配置的参数:--data
parser.add_argument('--data', ...)
可以直接修改,也可以修改形参:
右键代码打开修改运行参数
在形参里添加一个用于数据配置文件的路径的参数。
内容为训练集路径和测试集路径: