Cómo instalar Caffe en Ubuntu

Pasos para usar Caffe+MNIST para entrenar la red Lenet en una máquina Ubuntu 14.04 de 64 bits

1. Ubique la terminal en el directorio raíz de Caffe

2. Extraiga la base de datos MNIST: $ ./data/mnist/get_mnist.sh

3 Conviértala al formato de base de datos Lmdb: $ ./examples/mnist/create_mnist.sh

Ejecute esto. shell Después del script, ./examples/mnist/create_mnist.sh

agregará dos nuevos directorios, mnist_test_lmdb y mnist_train_lmdb, en ./examples/mnist/create_mnist.sh

. /examples/mnist/train_lenet.sh

(1) Utilice la red LeNet ("Aprendizaje basado en gradientes aplicado al reconocimiento de documentos"); / lenet_train_test.prototxtmodel;

(3), use ./examples/mnist/lenet_solver.prototxtmodel;

(4) Al ejecutar el script train_lenet.sh se llamará ./build/ directorio de herramientas El archivo ejecutable caffe se implementa como el archivo caffe.cpp en el directorio ./tools;

(5) Después de ejecutar este script, se generarán varios archivos, incluido ./examples /mnist/lenet_iter_10000 .caffemodel es el archivo de modelo generado por el entrenamiento final;

(6). El valor predeterminado anterior es ejecutarlo en modo GPU. Si desea que se ejecute en modo CPU, simplemente cambie lenet_solver. campo solver_mode en el archivo prototxt de la GPU a la CPU.