yolov5不加载预训练权重
时间: 2023-12-03 21:40:23 浏览: 430
yolov5 代码版 芒果检测 带训练权重,可以直接用
5星 · 资源好评率100%
如果您不想在yolov5中加载预训练权重,可以在训练脚本中设置--weights参数为空。这将导致模型使用随机初始化的权重进行训练,而不是使用预训练权重进行初始化。例如:
```shell
python train.py --img 640 --batch 16 --epochs 30 --data coco.yaml --weights ''
```
请注意,这可能会导致模型的训练时间更长,并且需要更多的数据来达到与使用预训练权重相同的性能水平。
阅读全文