Cargando…

Easy domain adaptation method for filling the species gap in deep learning-based fruit detection

Fruit detection and counting are essential tasks for horticulture research. With computer vision technology development, fruit detection techniques based on deep learning have been widely used in modern orchards. However, most deep learning-based fruit detection models are generated based on fully s...

Descripción completa

Detalles Bibliográficos
Autores principales: Zhang, Wenli, Chen, Kaizhen, Wang, Jiaqi, Shi, Yun, Guo, Wei
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Nature Publishing Group UK 2021
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8167097/
https://www.ncbi.nlm.nih.gov/pubmed/34059636
http://dx.doi.org/10.1038/s41438-021-00553-8
_version_ 1783701622601285632
author Zhang, Wenli
Chen, Kaizhen
Wang, Jiaqi
Shi, Yun
Guo, Wei
author_facet Zhang, Wenli
Chen, Kaizhen
Wang, Jiaqi
Shi, Yun
Guo, Wei
author_sort Zhang, Wenli
collection PubMed
description Fruit detection and counting are essential tasks for horticulture research. With computer vision technology development, fruit detection techniques based on deep learning have been widely used in modern orchards. However, most deep learning-based fruit detection models are generated based on fully supervised approaches, which means a model trained with one domain species may not be transferred to another. There is always a need to recreate and label the relevant training dataset, but such a procedure is time-consuming and labor-intensive. This paper proposed a domain adaptation method that can transfer an existing model trained from one domain to a new domain without extra manual labeling. The method includes three main steps: transform the source fruit image (with labeled information) into the target fruit image (without labeled information) through the CycleGAN network; Automatically label the target fruit image by a pseudo-label process; Improve the labeling accuracy by a pseudo-label self-learning approach. Use a labeled orange image dataset as the source domain, unlabeled apple and tomato image dataset as the target domain, the performance of the proposed method from the perspective of fruit detection has been evaluated. Without manual labeling for target domain image, the mean average precision reached 87.5% for apple detection and 76.9% for tomato detection, which shows that the proposed method can potentially fill the species gap in deep learning-based fruit detection.
format Online
Article
Text
id pubmed-8167097
institution National Center for Biotechnology Information
language English
publishDate 2021
publisher Nature Publishing Group UK
record_format MEDLINE/PubMed
spelling pubmed-81670972021-06-07 Easy domain adaptation method for filling the species gap in deep learning-based fruit detection Zhang, Wenli Chen, Kaizhen Wang, Jiaqi Shi, Yun Guo, Wei Hortic Res Article Fruit detection and counting are essential tasks for horticulture research. With computer vision technology development, fruit detection techniques based on deep learning have been widely used in modern orchards. However, most deep learning-based fruit detection models are generated based on fully supervised approaches, which means a model trained with one domain species may not be transferred to another. There is always a need to recreate and label the relevant training dataset, but such a procedure is time-consuming and labor-intensive. This paper proposed a domain adaptation method that can transfer an existing model trained from one domain to a new domain without extra manual labeling. The method includes three main steps: transform the source fruit image (with labeled information) into the target fruit image (without labeled information) through the CycleGAN network; Automatically label the target fruit image by a pseudo-label process; Improve the labeling accuracy by a pseudo-label self-learning approach. Use a labeled orange image dataset as the source domain, unlabeled apple and tomato image dataset as the target domain, the performance of the proposed method from the perspective of fruit detection has been evaluated. Without manual labeling for target domain image, the mean average precision reached 87.5% for apple detection and 76.9% for tomato detection, which shows that the proposed method can potentially fill the species gap in deep learning-based fruit detection. Nature Publishing Group UK 2021-06-01 /pmc/articles/PMC8167097/ /pubmed/34059636 http://dx.doi.org/10.1038/s41438-021-00553-8 Text en © The Author(s) 2021 https://creativecommons.org/licenses/by/4.0/Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made. The images or other third party material in this article are included in the article’s Creative Commons license, unless indicated otherwise in a credit line to the material. If material is not included in the article’s Creative Commons license and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this license, visit http://creativecommons.org/licenses/by/4.0/ (https://creativecommons.org/licenses/by/4.0/) .
spellingShingle Article
Zhang, Wenli
Chen, Kaizhen
Wang, Jiaqi
Shi, Yun
Guo, Wei
Easy domain adaptation method for filling the species gap in deep learning-based fruit detection
title Easy domain adaptation method for filling the species gap in deep learning-based fruit detection
title_full Easy domain adaptation method for filling the species gap in deep learning-based fruit detection
title_fullStr Easy domain adaptation method for filling the species gap in deep learning-based fruit detection
title_full_unstemmed Easy domain adaptation method for filling the species gap in deep learning-based fruit detection
title_short Easy domain adaptation method for filling the species gap in deep learning-based fruit detection
title_sort easy domain adaptation method for filling the species gap in deep learning-based fruit detection
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8167097/
https://www.ncbi.nlm.nih.gov/pubmed/34059636
http://dx.doi.org/10.1038/s41438-021-00553-8
work_keys_str_mv AT zhangwenli easydomainadaptationmethodforfillingthespeciesgapindeeplearningbasedfruitdetection
AT chenkaizhen easydomainadaptationmethodforfillingthespeciesgapindeeplearningbasedfruitdetection
AT wangjiaqi easydomainadaptationmethodforfillingthespeciesgapindeeplearningbasedfruitdetection
AT shiyun easydomainadaptationmethodforfillingthespeciesgapindeeplearningbasedfruitdetection
AT guowei easydomainadaptationmethodforfillingthespeciesgapindeeplearningbasedfruitdetection