1.pytorch训练好的模型在测试时出现如下错误
2.除了网上说的改为只加载参数外,以下方法亲测有用,在主运行代码中加入以下三行代码即可
import os,sys
root_path = os.getcwd()
sys.path.insert(0,root_path+"/yolov5") #这个后面的"/yolov5" 文件夹应该是指向的 yolov5根目录.
1.pytorch训练好的模型在测试时出现如下错误
2.除了网上说的改为只加载参数外,以下方法亲测有用,在主运行代码中加入以下三行代码即可
import os,sys
root_path = os.getcwd()
sys.path.insert(0,root_path+"/yolov5") #这个后面的"/yolov5" 文件夹应该是指向的 yolov5根目录.