Cómo utilizar Caffe para entrenar la red de clasificación ImageNet

1. Descargue el conjunto de datos de entrenamiento y el conjunto de datos de validación de ImageNet; guárdelo en el siguiente formato:

/path/to/imagenet/train/n01440764/n01440764_10026.JPEG

/path /to /imagenet/val/ILSVRC2012_val_00000001.JPEG

2. Realice algunas operaciones de preprocesamiento:

cd $CAFFE_ROOT/data/ilsvrc12/

./get_ilsvrc_aux. sh

3. Coloque los datos de entrenamiento y los datos de prueba en train.txt y val.txt respectivamente, que tienen sus archivos y etiquetas correspondientes

4. Coloque los nombres de clase de cada clase en 0--999 ​​y coloque sus nombres de categoría correspondientes en synset_words.txt para almacenar la relación de mapeo entre ellos.

5. El autor mencionó cómo hacerlo. Si todas las imágenes deben normalizarse a 256*256 primero, el autor también mencionó el uso de Mapreduce para acelerar este proceso.

O usted; Puede hacer esto directamente:

para el nombre en /path/to/imagenet/val/*.JPEG

convert -resize 256x256\! $nombre $nombre

Listo