提取yolov5的conv层数

以下是提取yolov5的conv层数的方法：

import torch
from models.yolo import Model

model = Model('models/yolov5s.yaml') # 加载yolov5s.yaml配置文件
state_dict = torch.load('models/yolov5s.pt', map_location='cpu')['model'].float().state_dict() # 加载yolov5s.pt模型参数
model.load_state_dict(state_dict, strict=False) # 加载模型参数

# 遍历模型的所有层，找到所有的conv层
for name, module in model.named_modules():
    if isinstance(module, torch.nn.Conv2d):
        print(name)

运行以上代码，即可输出yolov5模型中所有的conv层的名称。