IBM_AI/vi-opton/opton_od_pred/template/pred.bat

@echo off

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


set ROOT_DIR=%~dp0

set LOG_PATH=%ROOT_DIR%\log
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 CAFFEMODEL="%ROOT_DIR%/model/ZF_faster_rcnn_final.caffemodel"
set PROTOTXT="%ROOT_DIR%/model/prototxt/ZF/alt_opt/faster_rcnn_test.pt"

python pred.py ^
--force_cpu %FORCE_CPU% ^
--def %PROTOTXT% ^
--net %CAFFEMODEL% ^
--pred_dir %ROOT_DIR%/output/%TASK_NAME% ^
--img_dir %IMG_DIR% ^
--bbox_pred_margin 0 ^
--save_img True ^
--save_label True ^
--save_mark True ^
--splits_num 1 ^
--conf_thresh 0.1