如何在python中显示caffe层的所有名称？

Question

我正在创建一些自定义图层，以便我要检查caffe是否成功构建。 谁能告诉我如何显示python3中所有图层的名称

我正在使用python3.6并从caffe的最新版本进行构建

Answer 1

找到存储caffe python文件的目录 path_caffe ，您应该看到caffe文件夹classify.py和文件CMakeLists.txt，detect.py，draw_net.py，requirements.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)