OneFormer: One Transformer to Rule Universal Image Segmentation
Нам в компьютерном зрении давно нужен такой трансформер, который порешает все задачи за раз. Вот ступенька в этом направлении. OneFormer – это первый трансформер, который умеет решать три разных задачи сегментации : semantic, instance и panoptic segmentation.
Причем, он достигает SOTA на все трёх! Раньше нужно было тренировать три разные сетки для этого, а теперь – всего один OneFormer.
Чтобы указать модели, какую задачу сейчас решать, нужно просто подать нужный токен на вход. Удобненько.
❱❱ Код
❱❱ Arxiv
@ai_newz
Нам в компьютерном зрении давно нужен такой трансформер, который порешает все задачи за раз. Вот ступенька в этом направлении. OneFormer – это первый трансформер, который умеет решать три разных задачи сегментации : semantic, instance и panoptic segmentation.
Причем, он достигает SOTA на все трёх! Раньше нужно было тренировать три разные сетки для этого, а теперь – всего один OneFormer.
Чтобы указать модели, какую задачу сейчас решать, нужно просто подать нужный токен на вход. Удобненько.
❱❱ Код
❱❱ Arxiv
@ai_newz