Looking for caffe Answers? Try Ask4KnowledgeBase
Looking for caffe Keywords? Try Ask4Keywords

caffeCapas de Python personalizadas


Introducción

Este tutorial lo guiará a través de los pasos para crear una capa personalizada simple para Caffe usando python. Al final, hay algunos ejemplos de capas personalizadas. Por lo general, crearía una capa personalizada para implementar una funcionalidad que no está disponible en Caffe, ajustándola a sus requisitos.

La creación de una capa personalizada de Python agrega algo de sobrecarga a su red y probablemente no sea tan eficiente como una capa personalizada de C ++. Sin embargo, de esta manera, no tendrá que compilar todo el cafe con su nueva capa.

Parámetros

Parámetro Detalles
parte superior Una matriz con las mejores manchas de tu capa. Acceda a los datos que se le pasan usando top [i] .data , donde i es el índice de un blob específico
fondo Una matriz con las manchas inferiores de tu capa. Acceda a los datos que se le pasan usando la parte inferior [i] .data , donde i es el índice de un blob específico

Observaciones

- Caffe construir con la capa de Python

Caffe debe compilarse con la opción WITH_PYTHON_LAYER :

WITH_PYTHON_LAYER=1 make && make pycaffe

- ¿Dónde debo guardar el archivo de clase?

Tienes dos opciones (al menos que yo sepa). O bien, puede guardar el archivo de capa personalizado en la misma carpeta en la que va a ejecutar el comando caffe (probablemente donde se encuentren sus archivos de prototxt). Otra forma, también mi favorita, es guardar todas sus capas personalizadas en una carpeta y agregar esta carpeta a su PYTHONPATH.

Referencias

  1. El blog de Christopher Bourez.
  2. Caffe Github
  3. Desbordamiento de pila

Capas de Python personalizadas Ejemplos relacionados