RuntimeError: Could not open file SqueezeNet_deploy.prototxt

运行caffe代码时,加载模型prototxt文件:

import sysimport osimport numpy as npimport picklehelp_ = '''Usage:    decode.py <net.prototxt> <net.binary> <target.caffemodel>    Set variable CAFFE_ROOT as root of caffe before run this demo!'''if len(sys.argv) != 4:    print help_    sys.exit()else:    prototxt = sys.argv[1]    net_bin = sys.argv[2]    target = sys.argv[3]# os.system("cd $CAFFE_ROOT")caffe_root = os.environ["CAFFE_ROOT"]os.chdir(caffe_root)print caffe_rootsys.path.insert(0, caffe_root + 'python')import caffecaffe.set_mode_cpu()net = caffe.Net(prototxt, caffe.TEST)

输入参数设置为:

export CAFFE_ROOT=/home/caffe/

prototxt=SqueezeNet_deploy.prototxt

运行后出现错误:
RuntimeError: Could not open file SqueezeNet_deploy.prototxt
查看原因是因为prototxt文件路径需要为绝对路径,因此将输入改为:
prototxt=/home/SqueezeNet/SqueezeNet_deploy.prototxt
修改后,运行正确: