Interactive Image Segmentation Based on Label Pair Diffusion

Tao Wang,Shengzhe Qi,Jian Yang,Zexuan Ji,Quansen Sun,Qi Ge
DOI: https://doi.org/10.1109/tii.2020.2982995
IF: 12.3
2021-01-01
IEEE Transactions on Industrial Informatics
Abstract:This article explores the relationships between image element pairs and label pairs and extends label diffusion to label pair diffusion for the interactive image segmentation task. Compared with label diffusion, more accurate relationships between unlabeled and labeled data can be captured on a tensor product graph (TPG) by using higher order information, and more complex interactions among image elements and finer relationships between image element pairs and label pairs are explored in label pair diffusion (LPD) process. We first establish a prior label estimation framework to measure the label pair prior probability. Then, a probability learning process on TPG is designed to smooth the label prior. The learning process is equivalent to an iterative LPD process on the original graph, which makes the proposed algorithm maintain computational efficiency. Finally, the unary label probabilities can be obtained by a total-probability-theorem-based conversion from the binary relationships. Experiments on popular segmentation data sets demonstrate the superior performance of the proposed method.
automation & control systems,computer science, interdisciplinary applications,engineering, industrial
What problem does this paper attempt to address?