如需使用opencv来加载模型,则需将.pth转化为.onnx格式的模型。 a.先安装onnx,使用命令:pip install onnx; b.使用以下命令转为.onnx模型 importio importtorch importtorch.onnx importtorchvision frommodelsimportfcn device = torch.device("cuda:0"iftorch.cuda.is_available()else"cpu") deftest(): model...
报错原因: onnx的版本比较低,将onnx的版本太低了,将onnx的版本提高 执行指令,安装不注明版本号,自动安装最新版本号: AI检测代码解析 pip3 install onnx 1. 注明临时换源: AI检测代码解析 pip3 install onnx -i https://pypi.tuna.tsinghua.edu.cn/simple/ 1. 再进行.pth文件转换为onnx文件通用神经网络...
parser.add_argument("--checkpoint_model", default="yolov3_ckpt_69.pth",type=str, help="path to checkpoint model") opt = parser.parse_args() if not onnx_path.endswith('.onnx'): print('Warning! The onnx model name is not correct,\ please give a name that ends with \'.onnx\'!