yolov5-7.0更改为resnet主干网络

先确保自己可以正常运行coco128数据集,再去更改主干网络,避免后面出现不知名的报错。

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

目录

报错1:

步骤1:

步骤2:

步骤3:

步骤4:

步骤5:

运行成功:

win-resnet

ubuntu-resnet


191407

大神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&#
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值