先确保自己可以正常运行coco128数据集,再去更改主干网络,避免后面出现不知名的报错。
ClearML教程:https://blog.youkuaiyun.com/qq_40243750/article/details/126445671
目录
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&#