把yolov5 V6.0版本中根目录下的gen_wts.py文件复制到yolov5 V5.0版本的根目录中，执行导出.wts文件命令：
python3 gen_wts.py -w best.pt -o best.wts
报错：
Can't get attribute 'SPPF' on <module 'models.common，，，，，，
解决：
把yolov5 V6.0版本中的./models/common.py文件中的SPPF类复制到yolov5 V5.0版本的./models/common.py文件中，
并在common.py添加一行代码：
import warnings
再次导出.wts文件
python3 gen_wts.py -w best.pt -o best.wts
成功生成best.wts文件
O了！
posted on 2022-08-01 10:43
小王 阅读(811)
评论(0) 编辑 收藏 引用 所属分类:
TensorRT