mirror of
https://github.com/alibaba/EasyCV.git
synced 2025-06-03 14:49:00 +08:00
0128e881d9
* 将COCO_Stuff_164k数据集增加到了data_hub.md, prepared_data.md, 以及将其对应脚本加到了prepare_data文件夹下面 * 增加了segformer对应的predictor * 完善docs/source/model_zoo_seg.md中关于segformer的信息
8 lines
271 B
Python
8 lines
271 B
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
from .fcn_head import FCNHead
|
|
from .mask2former_head import Mask2FormerHead
|
|
from .segformer_head import SegformerHead
|
|
from .uper_head import UPerHead
|
|
|
|
__all__ = ['FCNHead', 'UPerHead', 'Mask2FormerHead', 'SegformerHead']
|