yolov5s冻结前10层,冻结的是那些层
时间: 2023-06-10 14:05:24 浏览: 102
这是属于yolov5s.pt的权重文件,放到根目录下
对于 yolov5s 模型,前 10 层一般包括以下层:
1. Conv2D(32, 3x3, stride=1, padding=1)
2. SiLU
3. Conv2D(64, 3x3, stride=2, padding=1)
4. SiLU
5. SPP
6. Conv2D(64, 1x1)
7. SiLU
8. Conv2D(64, 3x3, stride=2, padding=1)
9. SiLU
10. SPP
如果要冻结前 10 层,需要将这些层的参数设置为不可训练。
阅读全文