IBM_AI/vi-opton/opton_cls_train/pred.bat

@echo off

::##########################################################
::
:: ./pred.bat IMG_DIR [TASK_NAME] [cpu]
::
::##########################################################

REM set NET=bvlc_alexnet
set NET=bvlc_googlenet

set ROOT_DIR=%~dp0

set LOG_PATH=%ROOT_DIR%\log\%NET%
mkdir %LOG_PATH%
set LOG="%LOG_PATH%\pred_%date:~0,4%-%date:~5,2%-%date:~8,2%_%time:~0,2%-%time:~3,2%-%time:~6,2%.txt"
echo Logging output to %LOG%

set IMG_DIR=%1
set TASK_NAME=%2
set FORCE_CPU=0
if "%31"=="cpu" (set FORCE_CPU=1)

set SNAPSHOT_ITER=100000
set CAFFEMODEL="%ROOT_DIR%/output/snapshot/%NET%/%NET%_iter_%SNAPSHOT_ITER%.caffemodel"
set PROTOTXT="%ROOT_DIR%prototxt/%NET%/deploy.prototxt"

python pred.py ^
--force_cpu %FORCE_CPU% ^
--def %PROTOTXT% ^
--net %CAFFEMODEL% ^
--pred_dir %ROOT_DIR%/output/pred/%NET%/%SNAPSHOT_ITER%/%TASK_NAME% ^
--img_dir %IMG_DIR% ^
--save_img True ^
--save_mark True
REM --mean %ROOT_DIR%/data/images_mean.npy ^