跳到正文
原文
Hugging Face Blog·· 2023-01-19AI 评分40

Hugging Face transformers 集成 Mask2Former 与 OneFormer,统一图像分割

Universal Image Segmentation with Mask2Former and OneFormer

AI 导读

Mask2Former 和 OneFormer 两款图像分割模型现已集成到 Hugging Face 开源库 transformers 中。Mask2Former 统一了实例、语义和全景分割三种任务,但需分别训练;OneFormer 仅需在全景数据集上训练一次即可在三种任务上达到 SOTA,精度更高但因额外文本编码器延迟更大。作者已发布 30 个基于不同数据集训练的 checkpoint。

来源:Hugging Face Blog · huggingface.co