arXiv · 2205.09949
HCFormer: Unified Image Segmentation with Hierarchical Clustering
Abstract
Hierarchical clustering is an effective and efficient approach widely used for classical image segmentation methods. However, many existing methods using neural networks generate segmentation masks directly from per-pixel features, complicating the architecture design and degrading the interpretability. In this work, we propose a simpler, more interpretable architecture, called HCFormer. HCFormer accomplishes image segmentation by bottom-up hierarchical clustering and allows us to interpret, visualize, and evaluate the intermediate results as hierarchical clustering results. HCFormer can address semantic, instance, and panoptic segmentation with the same architecture because the pixel clustering is a common approach for various image segmentation tasks. In experiments, HCFormer achieves comparable or superior segmentation accuracy compared to baseline methods on semantic segmentation (55.5 mIoU on ADE20K), instance segmentation (47.1 AP on COCO), and panoptic segmentation (55.7 PQ on COCO).
Explore related subjects
Keep this discovery
Teppei Suzuki. 2022-05-20. HCFormer: Unified Image Segmentation with Hierarchical Clustering. https://arxiv.org/abs/2205.09949
Cite the original work for its findings. Save a collection to share your selection of sources.