yolov5训练后的权重文件路径在哪
时间: 2023-10-25 20:20:29 浏览: 554
手势数字识别 数据集以及权重文件 yolov5 (6.0) 格式txt 训练结果+权重文件
5星 · 资源好评率100%
在使用YOLOv5进行训练时,训练后的权重文件默认保存在项目根目录下的`runs/train/exp/weights`文件夹中,其中`train`是你的训练数据集名称,`exp`是你的实验名称。你也可以在训练命令中使用`--weights`参数来指定权重文件的保存路径。例如:
```
python train.py --data coco.yaml --cfg models/yolov5s.yaml --weights weights/yolov5s.pt --batch-size 64
```
这个命令将使用`coco.yaml`数据集配置文件和`yolov5s.yaml`模型配置文件进行训练,并将最终的权重文件保存在`weights/yolov5s.pt`路径下。
阅读全文