Шпаргалка для caffe / pycaffe?

кто-нибудь знает, есть ли шпаргалка для всех важных команд pycaffe? До сих пор я использовал caffe только через интерфейс Matlab и сценарии terminal + bash.

Я хотел перейти к использованию ipython и работать с примерами IPython notebook. Однако мне трудно получить обзор всех функций, которые находятся внутри модуля caffe для python. (Я также довольно новичок в python).

44
caffe ipython-notebook pycaffe

2 ответов

на тесты pycaffe и этот файл являются основным шлюзом к интерфейсу кодирования python.

прежде всего, вы хотели бы выбрать, следует ли использовать Caffe с CPU или GPU. Достаточно позвонить caffe.set_mode_cpu() или caffe.set_mode_gpu(), соответственно.

Net

основным классом, который предоставляет интерфейс pycaffe, является Net. Он имеет два конструктора:

net = caffe.Net('/path/prototxt/descriptor/file', caffe.TRAIN)

которые просто создают Net (в этом случае с помощью Слой Данных указано для обучения), или

net = caffe.Net('/path/prototxt/descriptor/file', '/path/caffemodel/weights/file', caffe.TEST)

создает Net и автоматически загружает веса, сохраненные в предоставленном caffemodel file-в этом случае с помощью Слой Данных указано для тестирования.

A или создание нового пласт тип. Эти функции используются реже, но их довольно легко понять, прочитав тестовые примеры.

95

обратите внимание, что ответ Флавио Феррара имеет небольшую проблему, которая может привести к потере много времени:

net.blobs['data'].data[...] = my_image
net.forward()

приведенный выше код не эффективен, если ваш первый слой является слоем типа данных, потому что когда net.forward() он начнется с первого слоя, и затем вставленный данных my_image будут покрыты. Так что это не ошибка, но совершенно не выход. Правильный способ-назначить начальный и конечный слои, для пример:

net.forward(start='conv1', end='fc')

вот репозиторий GitHub эксперимента по проверке лица на наборе данных LFW, используя pycaffe и некоторый код matlab. Я думаю, это может помочь много, особенно .

https://github.com/AlfredXiangWu/face_verification_experiment

кроме того, вот краткий пример кода использования pycaffe для изображения классификация:

http://codrspace.com/Jaleyhd/caffe-python-tutorial/ http://prog3.com/sbdm/blog/u011762313/article/details/48342495

10