как показать все названия слоев кофе в python? - PullRequest
0 голосов
/ 16 марта 2020

Я создаю несколько настраиваемых слоев, так что я хочу, чтобы оформить заказ кафе успешно или нет. Может кто-нибудь сказать мне, как показать все имена слоев в python3

Я использую python3 .6 и сборка из последней версии caffe

1 Ответ

1 голос
/ 18 марта 2020

Найдите каталог 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)
...