rename model

config/softgroup_default_scannet.yaml

@@ -1,7 +1,7 @@
 GENERAL:
   task: train  # train, test
   manual_seed: 123
-  model_dir: model/hais/hais.py
+  model_dir: model/softgroup/softgroup.py
   dataset_dir: data/scannetv2_inst.py
 
 DATA:
@@ -25,7 +25,7 @@ DATA:
   mode: 4 # 4=mean
 
 STRUCTURE:
-  model_name: hais
+  model_name: softgroup
   width: 32
   block_residual: True
   block_reps: 2

config/softgroup_fold5_backbone_s3dis.yaml

@@ -47,7 +47,7 @@ TRAIN:
   score_scale: 50 # the minimal voxel size is 2cm
   score_fullscale: 20  
   score_mode: 4 # mean
-  pretrain_path: 'softgroup_ckpt.pth'
+  pretrain_path: 'hais_ckpt.pth'
   pretrain_module: ['input_conv', 'unet', 'output_layer']
   fix_module: []
 

config/softgroup_fold5_default_s3dis.yaml

@@ -92,7 +92,7 @@ TEST:
   eval: True
   save_semantic: False
   save_pt_offsets: False
-  save_instance: True
+  save_instance: False
 
   test_mask_score_thre: -0.5 # bias fg << bg