Raster2Seq / raster2graph /config.json
haopt's picture
Upload folder using huggingface_hub
f89deec verified
{
"checkpoint_key": "raster2graph",
"name": "Raster2Graph",
"checkpoint_filename": "checkpoint.pth",
"source_local_filename": "r2g_sem_res256_ep0549.pth",
"metrics": {
"room_f1": 97.0
},
"inference_args": {
"dataset_name": "r2g",
"dataset_root": "data/R2G_hr_dataset_processed_v1",
"eval_set": "test",
"output_dir": "eval_outputs/r2g_sem_results",
"semantic_classes": 13,
"input_channels": 3,
"poly2seq": true,
"seq_len": 512,
"num_bins": 32,
"disable_poly_refine": true,
"dec_attn_concat_src": true,
"ema4eval": true,
"use_anchor": true,
"per_token_sem_loss": true,
"save_pred": true
}
}