IBM_AI/vi-opton/opton_cls_train/make_data_files.bat

@echo off

set CAFFE_HOME=D:\vi\packages\caffe-bvlc
set TOOLS=%CAFFE_HOME%\tools
set PYTHONPATH=%CAFFE_HOME%\python;%PYTHONPATH%

set RESIZE_HEIGHT=0
set RESIZE_WIDTH=0

set ROOT_DIR=%~dp0
set LOG=%ROOT_DIR%\data\make_data_files.log
echo Logging output to %LOG%

echo.
echo Clean up...

del %ROOT_DIR%\data\train_lmdb /f /s /q
del %ROOT_DIR%\data\val_lmdb /f /s /q
rd %ROOT_DIR%\data\train_lmdb /s /q
rd %ROOT_DIR%\data\val_lmdb /s /q
del %ROOT_DIR%\data\images_mean.binaryproto /f /s /q
del %ROOT_DIR%\data\*.txt /f /s /q

echo.
echo Create train.txt and val.txt...

python split_train_data.py ^
--train_percent 1.0 ^
--val_percent 0.1

echo.
echo Creating train lmdb...

%TOOLS%\convert_imageset.exe ^
    --resize_height=%RESIZE_HEIGHT% ^
    --resize_width=%RESIZE_WIDTH% ^
    --shuffle ^
    %ROOT_DIR%\data\dataset\ ^
    %ROOT_DIR%\data\train.txt ^
    %ROOT_DIR%\data\train_lmdb

echo.
echo Creating val lmdb...

%TOOLS%\convert_imageset.exe ^
    --resize_height=%RESIZE_HEIGHT% ^
    --resize_width=%RESIZE_WIDTH% ^
    --shuffle ^
    %ROOT_DIR%\data\dataset\ ^
    %ROOT_DIR%\data\train.txt ^
    %ROOT_DIR%\data\val_lmdb

echo.
echo Compute images mean...

%TOOLS%\compute_image_mean.exe %ROOT_DIR%\data\train_lmdb ^
    %ROOT_DIR%\data\images_mean.binaryproto

python save_images_mean.py %ROOT_DIR%\data\images_mean.binaryproto %ROOT_DIR%\data\images_mean.npy

echo.
echo Done.