yolov5-7.0更改resnet主干网络

参考链接

ClearML教程:https://blog.youkuaiyun.com/qq_40243750/article/details/126445671

b站教学视频：https://www.bilibili.com/video/BV1Mx4y1A7jy/spm_id_from=333.788&vd_source=b52b79abfe565901e6969da2a1191407

开始

github地址:https://github.com/z1069614715/objectdetection_script/tree/master

首先安装timm库：pip install -i https://pypi.tuna.tsinghua.edu.cn/simple timm

可以查看timm中包含的所有库：timm.list_models()

报错1:

解决:直接去这个路径下，删掉__init__.py就可以了，正常来说这是一个空文件，建议删除前检查一下

步骤1:

更改models中的yolov5.py文件

导入timm库

步骤2:

更改models中的yolov5.py文件

先改309行的parse_model函数

def parse_model(d, ch):  model_dict, input_channels(3)
    # Parse a YOLOv5 model.yaml dictionary
    LOGGER.info(f"\n{'':3}{'from':18}{'n':3}{'params':10}  {'module':<40}{'arguments':<30}")
    anchors, nc, gd, gw, act = d['anchors'], d['nc'], d['depth_multiple'], d['width_multiple'], d.get('activation')
    if act:
        Conv.default_act = eval(act)  redefine default activation, i.e. Conv.default_act = nn.SiLU()
        LOGGER.info(f"{colorstr('activation:')} {act}")  print
    na = (len(anchors[0]) // 2) if isinstance(anchors, list) else anchors  number of anchors
    no = na * (nc + 5)  number of outputs = anchors * (classes + 5)

    is_backbone = False
    layers, save, c2 = [], [], ch[-1]  layers, savelist, ch out
    for i, (f, n, m, args) in enumerate(d['backbone'] d['head']):  from, number, module, args
        try:
            t = m