diff --git a/configs/softgroup_stpls3d.yaml b/configs/softgroup_stpls3d.yaml index a54b248..e832bee 100644 --- a/configs/softgroup_stpls3d.yaml +++ b/configs/softgroup_stpls3d.yaml @@ -80,5 +80,5 @@ fp16: False epochs: 108 step_epoch: 20 save_freq: 4 -pretrain: './work_dirs/softgroup_stpls3d/latest.pth' +pretrain: './work_dirs/softgroup_stpls3d_backbone/latest.pth' work_dir: '' diff --git a/configs/softgroup_stpls3d_backbone.yaml b/configs/softgroup_stpls3d_backbone.yaml index a083b22..f0a92b8 100644 --- a/configs/softgroup_stpls3d_backbone.yaml +++ b/configs/softgroup_stpls3d_backbone.yaml @@ -7,6 +7,7 @@ model: semantic_only: True semantic_weight: [1.0, 1.0, 44.0, 21.9, 1.8, 25.1, 31.5, 21.8, 24.0, 54.4, 114.4, 81.2, 43.6, 9.7, 22.4] + with_coords: False ignore_label: -100 grouping_cfg: score_thr: 0.2