[notas de estudio de caffe] [01] [Convertir datos de imagen a archivo db]

1. Generar un archivo de lista de imágenes

toque create_filelist.sh

vim create_filelist.sh

 

1 #! / Usr / bin / bash
 2  
3 DATA = train # Recorre el directorio train o prueba
 4  
5  echo  " Create train.txt ... " 
6  rm -rf $ DATA / $ DATA.txt #Elimina la lista de archivos original
 7  
8  para  dir  en ` ls $ DATA` # Recorre cada carpeta bajo el directorio
 9  do 
10      if [-d $ DATA / $ dir ] # Si es una carpeta, extraiga la ruta del archivo debajo de la carpeta más el nombre de la carpeta Escriba el archivo de la lista de archivos
 11 y      luego 
12          encuentre $ DATA / $ dir -name * .png | cut -d ' / '-f3 | sed  " s / ^ / $ dir \ / & / g; s / $ / & $ dir / g " >> $ DATA / $ DATA.txt
 13      fi 
14  done 
15  echo  " Hecho ... "

 

2. Convertir imagen a base de datos

toque create_lmdb.sh

vim create_lmdb.sh

1 #! / Usr / bin / bash
 2  
3 DATA = train #Folder name
 4 LMDB = train_lmdb #Database name 5 HEIGHT = 28 #
                                                  Altura de la imagen
 6 WIDTH = 28 #                                                   Ancho de la imagen
 7 BUILD = / home / yuandanfei / caffe / build / tools / convert_imageset #Imagen a la ruta de la base de datos
 8 ROOTFOLDER = / home / yuandanfei / work / caffe / mnist / data #Image ruta raíz
 9  
10  rm - rf $ LMDB
 11  
12$ BUILD - barajar \
 13 --resize_height = $ {HEIGHT} \
 14 --resize_width = $ {WIDTH} \
 15 $ ROOTFOLDER / $ DATA / \
 16 $ ROOTFOLDER / $ DATA / $ DATA.txt $ LMDB

 

Materiales de referencia:

https://www.cnblogs.com/denny402/p/5082341.html

https://www.cnblogs.com/linux-wangkun/p/5745584.html