English

Model Doctor for Diagnosing and Treating Segmentation Error

Computer Vision and Pattern Recognition 2023-02-24 v2

Abstract

Despite the remarkable progress in semantic segmentation tasks with the advancement of deep neural networks, existing U-shaped hierarchical typical segmentation networks still suffer from local misclassification of categories and inaccurate target boundaries. In an effort to alleviate this issue, we propose a Model Doctor for semantic segmentation problems. The Model Doctor is designed to diagnose the aforementioned problems in existing pre-trained models and treat them without introducing additional data, with the goal of refining the parameters to achieve better performance. Extensive experiments on several benchmark datasets demonstrate the effectiveness of our method. Code is available at \url{https://github.com/zhijiejia/SegDoctor}.

Keywords

Cite

@article{arxiv.2302.08980,
  title  = {Model Doctor for Diagnosing and Treating Segmentation Error},
  author = {Zhijie Jia and Lin Chen and Kaiwen Hu and Lechao Cheng and Zunlei Feng and Mingli Song},
  journal= {arXiv preprint arXiv:2302.08980},
  year   = {2023}
}
R2 v1 2026-06-28T08:42:54.742Z