Back to Search Start Over

Model Doctor for Diagnosing and Treating Segmentation Error

Authors :
Jia, Zhijie
Chen, Lin
Hu, Kaiwen
Cheng, Lechao
Feng, Zunlei
Song, Mingli
Publication Year :
2023

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}.

Details

Database :
arXiv
Publication Type :
Report
Accession number :
edsarx.2302.08980
Document Type :
Working Paper