File size: 597 Bytes
fadb92b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
DATA=data/coco_cubicasa5k_nowalls_v4-1_refined
SPLIT=test 

NAME=r2g_cc5k_${SPLIT}_preds
SAVE_DIR=cross_eval_outputs/${NAME}
CKPT=checkpoints/r2g_sem_res256_ep0549.pth

python eval.py \
   --dataset_name=cubicasa \
   --dataset_root=${DATA}/ \
   --eval_set=${SPLIT} \
   --checkpoint=${CKPT} \
   --output_dir=${SAVE_DIR} \
   --semantic_classes=13 \
   --input_channels 3 \
   --poly2seq \
   --seq_len 512 \
   --num_bins 32 \
   --disable_poly_refine \
   --dec_attn_concat_src \
   --ema4eval \
   --use_anchor \
   --per_token_sem_loss \
   --save_pred \
   --disable_sem_rich \
   --drop_wd