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

文献类型： 外文期刊

第一作者： Zhang, Wenli

作者： Zhang, Wenli;Chen, Kaizhen;Wang, Jiaqi;Shi, Yun;Guo, Wei

作者机构：

期刊名称：HORTICULTURE RESEARCH （ 影响因子：6.793； 五年影响因子：6.589 ）

ISSN： 2662-6810

年卷期： 2021 年 8 卷 1 期

页码：

收录情况： SCI

摘要： 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.

分类号：

  • 相关文献
作者其他论文 更多>>

微信小程序