raster2seq / tools /eval_s3d_pretrain.sh
anas
Initial deployment of Raster2Seq floor plan vectorization API
fadb92b
#!/usr/bin/env bash
DATA=data/coco_s3d_bw/
FOLDER=test
CKPT=checkpoints/s3dbw_res256_ep1349.pth
python eval.py \
--dataset_name=stru3d \
--dataset_root=${DATA} \
--eval_set=${FOLDER} \
--checkpoint=${CKPT} \
--output_dir=eval_outputs/s3dbw_results/ \
--semantic_classes=-1 \
--input_channels=3 \
--poly2seq \
--seq_len 512 \
--num_bins 32 \
--disable_poly_refine \
--dec_attn_concat_src \
--use_anchor \
--ema4eval \
--save_pred