把yolov5 V6.0版本中根目錄下的gen_wts.py文件復(fù)制到y(tǒng)olov5 V5.0版本的根目錄中，執(zhí)行導(dǎo)出.wts文件命令：
python3 gen_wts.py -w best.pt -o best.wts
報(bào)錯(cuò)：
Can't get attribute 'SPPF' on <module 'models.common，，，，，，
解決：
把yolov5 V6.0版本中的./models/common.py文件中的SPPF類復(fù)制到y(tǒng)olov5 V5.0版本的./models/common.py文件中，
并在common.py添加一行代碼：
import warnings
再次導(dǎo)出.wts文件
python3 gen_wts.py -w best.pt -o best.wts
成功生成best.wts文件
O了！
posted on 2022-08-01 10:43
小王 閱讀(1014)
評(píng)論(0) 編輯 收藏 引用 所屬分類:
TensorRT