Mobile QR Code
Title A Matching-loss for Improving Deep Image Segmentation
Authors 문구영(Goo-Young Moon) ; 김종옥(Jong-Ok Kim)
DOI https://doi.org/10.5573/ieie.2024.61.6.75
Page pp.75-81
ISSN 2287-5026
Keywords Semantic segmentation; Small object detection; Mathing-loss; Crop images; Disease
Abstract Recently, in the filed of smart farm, there have been actively conducted studies to accurately detect disease regions by using segmentation for crop images. However, due to the small size of disease, predicting disease regions accurately is a challenging task. To address this issue, this paper proposes a novel optimization strategy, called Matching-Loss which stems from the translation invariance property of the segmentation network. Transformation invariance means that the feature embedding representation of the network for a transformed image should be identical to the transformed version of the feature embedding of the original input image. Matching loss offers the advantage of being easily calculated using Mean Squared Error (MSE). Furthermore, by leveraging the optimization based on the relationship between the input image and its transformed version, it is possible to enhance the generalization capability and robustness of the segmentation network. This improvement can lead to an enhanced recognition ability of the network for regions affected by various factors such as pests and diseases. We verified the significance of the matching loss through comparative experiments conducted on crop images captured in real field environments.