SegFormer 由分层 Transformer 编码器和轻量级的全 MLP 解码头组成,能在 ADE20K 和 Cityscapes 等语义分割基准数据集上取得出色效果。分层 Transformer 首先在 ImageNet-1k 上进行了预训练,然后添加了解码头,并在下游数据集上进行了微调。
Go to file
YYJ-aaaa 6088f43e38 first commit 2024-11-15 15:03:32 +08:00
.gitattributes Add .gitattributes 2024-11-15 14:25:06 +08:00
README.md Initial commit 2024-11-15 14:25:06 +08:00
config.json first commit 2024-11-15 15:03:32 +08:00
preprocessor_config.json first commit 2024-11-15 15:03:32 +08:00
pytorch_model.bin first commit 2024-11-15 15:03:32 +08:00
tf_model.h5 first commit 2024-11-15 15:03:32 +08:00

README.md

mit-b2_a13603898159394816634986

SegFormer 由分层 Transformer 编码器和轻量级的全 MLP 解码头组成,能在 ADE20K 和 Cityscapes 等语义分割基准数据集上取得出色效果。分层 Transformer 首先在 ImageNet-1k 上进行了预训练,然后添加了解码头,并在下游数据集上进行了微调。