Interactive Image Segmentation Based on Label Pair Diffusion

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.