get_flops.py 不支持mask2former

open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

https://mmsegmentation.readthedocs.io/en/main/

Apache License 2.0

7.69k stars 2.53k forks source link

get_flops.py 不支持mask2former #3645

Open Deom-z opened 2 months ago

Deom-z commented 2 months ago

if cfg.model.decode_head.type in ['MaskFormerHead', 'Mask2FormerHead']:
    # TODO: Support MaskFormer and Mask2Former
    raise NotImplementedError('MaskFormer and Mask2Former are not '
                              'supported yet.')

Saillxl commented 1 month ago

I also encountered the same problem. Have you solved it？

sym330 commented 1 day ago

same