bigdetection icon indicating copy to clipboard operation
bigdetection copied to clipboard

COCO fine-tuning without panoptic stuffthingmaps

Open titusss opened this issue 1 year ago • 0 comments

Hello thank you for the great work,

I want to fine-tune the pre-trained CBnet on a COCO-like custom dataset. However, the htc_cbv2_swin_base_adamw_20e_coco.py config expects COCO-stuff pixelmaps. (See configs/BigDetection/cbnetv2/htc_cbv2_swin_base_adamw_20e_coco.py)

In line 92 it tries to load stuffthingmaps into the data dict seg_prefix=data_root + 'stuffthingmaps/train2017/',

The whole code block, starting at line 89:

samples_per_gpu=1
data = dict(samples_per_gpu=samples_per_gpu,
            train=dict(
                seg_prefix=data_root + 'stuffthingmaps/train2017/',
                pipeline=train_pipeline),
            val=dict(pipeline=test_pipeline),
            test=dict(pipeline=test_pipeline))
optimizer = dict(_delete_=True, type='AdamW', lr=0.0001*(samples_per_gpu/2), betas=(0.9, 0.999), weight_decay=0.05,
                 paramwise_cfg=dict(custom_keys={'absolute_pos_embed': dict(decay_mult=0.),
                                                 'relative_position_bias_table': dict(decay_mult=0.),
                                                 'norm': dict(decay_mult=0.)}))

What if I do not want to train panoptic stuffthingmaps? I only want to train on train and val COCO sets. If I set seg_prefix=None, it will fail as it expects a string or path. How do I disable loading these panoptic thing maps, all together?

I am not sure if it's even possible to train native COCO (not COCO-stuff) without fixing that issue.

titusss avatar Sep 23 '22 20:09 titusss