Toybrick

标题: ---best.pt模型转换出错了！！！ [打印本页]

作者: linuxdog 时间: 2022-3-21 12:02
标题: ---best.pt模型转换出错了！！！

1. cd ..../rknn-toolkit2-1.0.0_no_docker/examples/pytorch/resnet18
$ python3 test.py
没有问题

2.  修改了test.py
把模型修改成:
      model = './best.pt'
3.  $ python3 test.py
现在报如下错误：

--> config model
done
--> Loading model
Traceback (most recent call last):
  File "test.py", line 66, in <module>
ret = rknn.load_pytorch(model=model, input_size_list=input_size_list)
  File "/home/rknn2/.local/lib/python3.6/site-packages/rknn/api/rknn.py", line 175, in load_pytorch
ret = self.rknn_base.load_pytorch(model=model, input_size_list=input_size_list, inputs=inputs, outputs=outputs)
  File "rknn/api/rknn_base.py", line 1882, in rknn.api.rknn_base.RKNNBase.load_pytorch
  File "/home/rknn2/.local/lib/python3.6/site-packages/torch/jit/__init__.py", line 275, in load
cpp_module = torch._C.import_ir_module(cu, f, map_location, _extra_files)
RuntimeError: [enforce fail at inline_container.cc:209] . file not found: archive/constants.pkl

4.  best.pt是从：yolov5目录训练完了拷贝出来的：
                              ./runs/train/exp5/weights/best.pt
   yolov5:    https://github.com/airockchip/yolov5

作者: tiger_k 时间: 2022-3-21 19:58
你这个是在板子上执行的吗还是PC机上

欢迎光临 Toybrick (https://t.rock-chips.com/)