遇到该问题时,百度了一下,没找到任何解决方案,谷歌了一下也没找到好的方案,于是决心要解决掉它。。。。
在配置yolov5(yolov5)时出现了以下问题:
AttributeError: Can’t get attribute ‘C3’ on <module ‘models.common’ from ‘/root/deepstream_yolov5/yolov5/models/common.py’>
具体:
解决方案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')