Найдите каталог path_caffe , где хранятся python файлы caffe, вы должны увидеть папку caffe classify.py и файлы CMakeLists.txt, detect.py, draw _net .py, require.txt и train.py. Затем найдите файл .prototxt вашей модели path_model . Наконец замените следующий код:
path_caffe = "path/to/caffe/python"
import sys
sys.path.append(path_caffe)
import caffe
path_model = "path/to/model.prototxt"
net = caffe.Net(path_model, caffe.TEST)
for key in net.layer_dict:
print(key)