An Improved Variational-Based Model for Denoising and Segmentation of Vector-Valued Images

Abstract

Preserving important features such as edges is one of the main concerns in models for denoising and segmenting vector-valued (colour) images. The Rudin-Osher-Fatemi (ROF) model is a well-known variational-based image denoising model that is capable of reducing image noise while preserving image edges. However, the ROF model is not formulated for denoising colour images and is less effective in preserving corners and weak edges. On the other hand, a variational-based selective segmentation model for colour images called the selective distance segmentation (DSS2) model has recently been proposed, which can effectively partition or extract a specific object in an image. However, the DSS2 model has problems in segmenting colour images with noise, which may result in poor segmentation. Therefore, in this research, we first modify the ROF model to denoise vector-valued images by including the edge detector and extending the formulation into a vector-valued framework. Second, we reformulate the DSS2 model by incorporating the modified ROF model as a new fitting term in the DSS2 model. Peak signal-to-noise ratio (PSNR) is used to measure the image quality, while Jaccard and Dice similarity index are used to evaluate the segmentation quality. The comparison between our proposed model and existing model shows that our model is more effective as indicated by higher PSNR, Jaccard and Dice similarity index values.