¡Cómo compilar CAFFE rápida y fácilmente en Windows y usar su interfaz matlab!
Preparé todo lo necesario, incluida la biblioteca de terceros que creé yo mismo, creé una carpeta de terceros en el directorio raíz del código fuente y descomprimí el archivo.
Después de descomprimir, agregue la carpeta 3rdparty/bin en el directorio del sistema para que el programa pueda encontrar las DLL de estas bibliotecas de terceros.
Luego está Visual Studio 2012, porque mi biblioteca de terceros está compilada en el entorno VS2012, por lo que si está utilizando otras versiones de VS, compílela usted mismo.
Finalmente, están CUDA y MKL. Puede solicitarlo oficialmente a Intel si no usa el modo CPU, en realidad no importa. paquete, también proporciono el archivo de biblioteca openblas.
En segundo lugar, compilar
La compilación es muy simple y se divide en los siguientes pasos:
1. Haz doble clic en ./src/caffe/proto/extract_proto. .bat por lotes Procese el archivo y genere dos archivos c++, caffe.pb.h y caffe.pb.cc, así como el archivo caffe_pb2.py utilizado por Python.
2. Abra ./build/MSVC/MainBuilder.sln y luego cambie el modo de compilación al modo Versión X64. Si muestra que la carga falló después de abrir, puede ser que su versión de CUDA no sea consistente con la mía. La mía es CUDA 6.5. En este momento, necesita usar el Bloc de notas para abrir ./build/MSVC/MainBuilder.vcxproj. CUDA 6.5, y use el suyo. Reemplace este 6.5 con la versión CUDA y luego podrá abrirlo normalmente.
3. Haga clic derecho en el proyecto MainCaller a la izquierda, seleccione "Propiedades", en la pestaña "C/C++", cambie el directorio de inclusión de cuda en el directorio "Inclusiones adicionales" a su propio directorio; En la pestaña "Enlazador", cambie el directorio de la biblioteca cuda en el directorio "Bibliotecas adicionales" a su propio directorio.
4. Haz clic en el triángulo verde encima de la barra de herramientas para compilar.
Si desea utilizar el contenedor matlab para extraer características y observar los pesos después del entrenamiento, simplemente cambie los directorios cuda y matlab en el proyecto matcaffe a sus propios directorios y luego compile, lo tendrá en matlab. Archivo /caffe Busque un archivo llamado caffe.mexw64 en la carpeta.
El Wrapper de Python es similar. Cambie el directorio en el proyecto pycaffe a su propio directorio y luego podrá generar un archivo dll de Python llamado _caffe.pyd en la carpeta python/caffe.