yolov5 1.0设置freeze
时间: 2023-10-07 15:08:48 浏览: 121
基于Yolov5 + onnx + PyQt5 + 的目标检测打包部署.zip
5星 · 资源好评率100%
在 YOLOv5 中,你可以通过在 `model.yaml` 文件中设置 `backbone.freeze_at` 参数来冻结模型的某些层。该参数表示冻结骨干网络(backbone)中前多少层。
例如,如果你想要冻结前 3 层,可以在 `model.yaml` 文件中添加以下内容:
```
backbone:
...
freeze_at: 3
```
这将导致骨干网络中的前 3 层被冻结,只有后面的层会被训练。这对于微调预训练的模型非常有用,以便在新的数据集上进行训练。
阅读全文