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

minor

Browse Source
This commit is contained in:
Thang Vu 2022-04-11 02:03:29 +00:00
parent c68be64a35
commit 08eb82816b
4 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
configs/softgroup_scannet.yaml
View File

@ -66,8 +66,9 @@ optimizer:
type: 'Adam'
lr: 0.004
fp16: False
epochs: 128
step_epoch: 50
save_freq: 4
pretrain: 'work_dirs/softgroup_scannet_backbone_spconv2_dist/epoch_116.pth'
work_dir: 'work_dirs/softgroup_scannet_spconv2_dist'
pretrain: 'work_dirs/softgroup_scannet_backbone/epoch_120.pth'
work_dir: ''

2
configs/softgroup_scannet_backbone.yaml
View File

@ -71,4 +71,4 @@ epochs: 128
step_epoch: 50
save_freq: 4
pretrain: ''
work_dir: 'work_dirs/softgroup_scannet_backbone'
work_dir: ''

5
requirements.txt
View File

@ -1,9 +1,10 @@
cmake>=3.13.2
munch
pandas
plyfile
pyyaml==5.4.1
scikit-learn
scipy
six
tensorboard
tensorboardX
torch==1.1
tqdm

3
test.py
View File

@ -33,10 +33,9 @@ if __name__ == '__main__':
cfg = Munch.fromDict(yaml.safe_load(cfg_txt))
logger = get_root_logger()
model = SoftGroup(**cfg.model)
model = SoftGroup(**cfg.model).cuda()
logger.info(f'Load state dict from {args.checkpoint}')
load_checkpoint(args.checkpoint, logger, model)
model.cuda()
dataset = build_dataset(cfg.data.test, logger)
dataloader = build_dataloader(dataset, training=False, **cfg.dataloader.test)