CR-Net / test_scripts /ast_n2h.sh
datnguyentien204's picture
Upload 147 files
0f52c9d verified
#!/bin/bash
set -e
EXPERIMENT_NAME='ast_rafael_v1_sharpening'
IMAGE_FOLDER='/home/share/RAFAEL_8/VOC_2012_dark_YOLO_format/images/trainval'
WHICH_EPOCH='latest'
MAX_IMG_HEIGHT=960
GPU_ID="0"
#@HOURS="16.0"
HOURS="16.0 17.0 18.0 19.0 20.0 21.0 22.0"
#HOURS="17.0 17.5 18.0 18.5 19.0 19.5 20.0"
ADD_MARK=false
RESULTS_DIR='./results_v2'
CHECKPOINTS_DIR='./checkpoints_v2'
TIMESTAMP_FLAG=""
if [ "$ADD_MARK" = true ]; then
TIMESTAMP_FLAG="--add_timestamp"
fi
python test.py \
--name "${EXPERIMENT_NAME}" \
--image_dir "${IMAGE_FOLDER}" \
--checkpoints_dir "${CHECKPOINTS_DIR}" \
--results_dir "${RESULTS_DIR}" \
--which_epoch "${WHICH_EPOCH}" \
--gpu_ids "${GPU_ID}" \
--hours ${HOURS} \
--load_from_opt_file \
--max_height "${MAX_IMG_HEIGHT}" \
${TIMESTAMP_FLAG}
echo "Testing finished. Results are in ${RESULTS_DIR}/${EXPERIMENT_NAME}/"