关于配置YOLOV5时出现错误 AttributeError: Can‘t get attribute ‘C3‘ on <module ‘models.common‘ from ‘/root/deep

遇到该问题时,百度了一下,没找到任何解决方案,谷歌了一下也没找到好的方案,于是决心要解决掉它。。。。

在配置yolov5(yolov5)时出现了以下问题:
AttributeError: Can’t get attribute ‘C3’ on
具体:

解决方案1:
找了各种办法,也去看了,没有什么具体的解决办法,最后是通过重新在别的文件夹或者路径下重新clone项目在运行来解决的问题,具体为什么,没搞明白。

我用的gen_wts.py如下:

import torch
import struct
from utils.torch_utils import select_device
import pickle

# Initialize
device = select_device('0')
# Load model
model = torch.load('weights/yolov5s.pt', map_location=device)['model'].float()  # load to FP32
model.to(device).eval()

with open('yolov5s.wts', 'w') as f:
    f.write('{}\n'.format(len(model.state_dict().keys())))
    for k, v in model.state_dict().items():
        vr = v.reshape(-1).cpu().numpy()
        f.write('{} {} '.format(k, len(vr)))
        for vv in vr:
            f.write(' ')
            f.write(struct.pack('>f',float(vv)).hex())
        f.write('\n')

解决方案2:

torch.load(‘weights/yolov5s.pt’, map_location=device)[‘model’].float() 改为 torch.load(‘weights/yolov5s.pt’, map_location=device)[‘model’].float(),
同时记得把yolov5/weights目录里面的yolov5s.pt(其他的权重以此类推)拷贝到yolov5目录下,重新运行python3 gen_wts.py即可。
gen_wts.py如下:

import torch
import struct
from utils.torch_utils import select_device
import pickle

# Initialize
device = select_device('0')
# Load model
model = torch.load('yolov5s.pt', map_location=device)['model'].float()  # load to FP32
model.to(device).eval()

with open('yolov5s.wts', 'w') as f:
    f.write('{}\n'.format(len(model.state_dict().keys())))
    for k, v in model.state_dict().items():
        vr = v.reshape(-1).cpu().numpy()
        f.write('{} {} '.format(k, len(vr)))
        for vv in vr:
            f.write(' ')
            f.write(struct.pack('>f',float(vv)).hex())
        f.write('\n')

你可能感兴趣的:(机器学习&深度学习,开发,GitHub,深度学习)