EasyCV/easycv/models/segmentation/heads/__init__.py

# Copyright (c) Alibaba, Inc. and its affiliates.
from .fcn_head import FCNHead
from .mask2former_head import Mask2FormerHead
from .segformer_head import SegformerHead
from .stdc_head import STDCHead
from .uper_head import UPerHead

__all__ = [
    'FCNHead', 'UPerHead', 'Mask2FormerHead', 'SegformerHead', 'STDCHead'
]
[feature] add segmentation modules (#71) * add segmentation module & add ResNetV1c & ResNetV1d & add benchmark 2022-05-31 20:19:17 +08:00			`# Copyright (c) Alibaba, Inc. and its affiliates.`
			`from .fcn_head import FCNHead`
add mask2former algo (#115) add mask2former algo support panopitc pipeline add segment predictor 2022-08-08 18:17:01 +08:00			`from .mask2former_head import Mask2FormerHead`
add segformer algo * 将COCO_Stuff_164k数据集增加到了data_hub.md, prepared_data.md, 以及将其对应脚本加到了prepare_data文件夹下面 * 增加了segformer对应的predictor * 完善docs/source/model_zoo_seg.md中关于segformer的信息 2022-08-18 10:40:18 +08:00			`from .segformer_head import SegformerHead`
[features] Support stdc (#284) * add stdc semantic segmentation algorithm 2023-02-16 14:00:59 +08:00			`from .stdc_head import STDCHead`
add upernet algo (#118) * add upernet algo * fix import onnx bug 2022-07-12 11:14:09 +08:00			`from .uper_head import UPerHead`
[feature] add segmentation modules (#71) * add segmentation module & add ResNetV1c & ResNetV1d & add benchmark 2022-05-31 20:19:17 +08:00
[features] Support stdc (#284) * add stdc semantic segmentation algorithm 2023-02-16 14:00:59 +08:00			`__all__ = [`
			`'FCNHead', 'UPerHead', 'Mask2FormerHead', 'SegformerHead', 'STDCHead'`
			`]`