欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Hi3559 Ruyi Studio ImportError: No module named 'rpn' Failed to get shape info.

程序员文章站 2022-03-06 21:33:11
...

按照《HISVP开发指南》文档配置好了环境,可是加载模型还是无法显示shape信息
ruyi studio报错如下

Traceback (most recent call last):
  File "E:\HISVP\RuyiStudio-2.0.38\Resources\pythonScript\get_caffe_shape.py", line 3, in <module>
    import rpn
ImportError: No module named 'rpn'
Failed to get shape info.
Failed to obtain the shape information. The possible causes are as follows: (1) The layers that the prototxt file contains are not supported by the tool. (2) The PC is not installed with the Python environment. (3) The PC is not installed with the Caffe environment. For details, see the following document: ruyi_env_setup-x.x.x/readme.txt.

前面还一直去检查环境变量的配置,确认和文档无异后还是不行。
最后原因是rpn在roi_pooling目录,sys.path没有这个路径,所以修改E:\HISVP\RuyiStudio-2.0.38\Resources\pythonScript\get_caffe_shape.py文件

import sys, getopt
sys.path.append('E:\\HISVP\\ruyi_env_setup-2.0.38\\python35\\Lib\\site-packages\\caffe\\roi_pooling')
import rpn

把import rpn挪到import sys后面,
import sys后加一句sys.path.append(‘E:\HISVP\ruyi_env_setup-.0.38\python35\Lib\site-packages\caffe\roi_pooling’)

这样再执行Mark shape信息就出来了
Hi3559 Ruyi Studio ImportError: No module named 'rpn' Failed to get shape info.

相关标签: 踩坑记录