Title

Score-based mask edge improvement of mask-RCNN for segmentation of fruit and vegetables

Document Type

Journal Article

Publication Title

Expert Systems with Applications

Volume

190

Publisher

Elsevier

School

School of Engineering

Funders

Edith Cowan University

Higher Education Comission (HEC)

Pakistan Islamia University of Bahawalpur

Comments

Hameed, K., Chai, D., & Rassau, A. (2022). Score-based mask edge improvement of mask-RCNN for segmentation of fruit and vegetables. Expert Systems with Applications, 190, article 116205.

https://doi.org/10.1016/j.eswa.2021.116205

Abstract

Machine intelligence based automation plays a significant role in many modern applications, and vision based understanding is a significant element of this. To meet the goals of vision based understanding of images, segmentation plays a vital role through partitioning of regions of interest for further processing. Much research, including basic statistical and modern convolutional neural network based techniques, has been reported for segmentation. However, application based fine tuning is always essential for effective results in complex applications. In this paper, we have proposed a score-based mask edge improvement of Mask-RCNN to segment fruit and vegetable images in a supermarket environment. A modular score-based edge improvement head is proposed for Mask-RCNN to improve the segmentation of fruit and vegetable images. The edge difference between ground truth and estimated edges is filled with a pre-defined score proportional to the pixel-level difference. A cosine similarity based loss function is reduced to improve the edge details following segmentation. A significant improvement has been reported based on the proposed technique.

DOI

10.1016/j.eswa.2021.116205

Access Rights

subscription content

Share

 
COinS