Toybrick

---best.pt模型转换出错了!!!

linuxdog

注册会员

积分
82
发表于 2022-3-21 12:02:57    查看: 1947|回复: 1 | [复制链接]    打印 | 显示全部楼层

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

注册会员

积分
89
发表于 2022-3-21 19:58:05 | 显示全部楼层
你这个是在板子上执行的吗 还是PC机上
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表