eval.sh 695 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
4
5
6
7
8
9
10
11
# !/bin/bash

set -x

JOB_NAME="painter_vit_large"
CKPT_FILE="painter_vit_large.pth"
PROMPT="study_room_0005b/rgb_00094"

MODEL="painter_vit_large_patch16_input896x448_win_dec64_8glb_sl1"
CKPT_PATH="models/${JOB_NAME}/${CKPT_FILE}"
DST_DIR="models_inference/${JOB_NAME}/nyuv2_depth_inference_${CKPT_FILE}_${PROMPT}"
chenych's avatar
chenych committed
12
DATA_DIR="datasets"
chenych's avatar
chenych committed
13
14
15
16
17
18
# inference
python eval/nyuv2_depth/painter_inference_depth.py \
  --ckpt_path ${CKPT_PATH} --model ${MODEL} --prompt ${PROMPT}

python eval/nyuv2_depth/eval_with_pngs.py \
  --pred_path ${DST_DIR} \
chenych's avatar
chenych committed
19
20
  --gt_path ${DATA_DIR}/nyu_depth_v2/official_splits/test/ \
  --data_dir ${DATA_DIR} \
chenych's avatar
chenych committed
21
  --dataset nyu --min_depth_eval 1e-3 --max_depth_eval 10 --eigen_crop