按照《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信息就出来了