vhad/SoftGroup
1
0
Fork 0
mirror of https://github.com/botastic/SoftGroup.git synced 2025-10-16 11:45:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

rename model

Browse Source
This commit is contained in:
Thang Vu 2022-03-12 05:51:08 +00:00
parent 85bb5e8dbd
commit a14122aa4b
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
config/softgroup_default_scannet.yaml
View File

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

2
config/softgroup_fold5_backbone_s3dis.yaml
View File

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

2
config/softgroup_fold5_default_s3dis.yaml
View File

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