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á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 |
Caffe debe compilarse con la opción WITH_PYTHON_LAYER
:
WITH_PYTHON_LAYER=1 make && make pycaffe
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.