Wheat Head Detection using Deep, Semi-Supervised and Ensemble Learning

Fares Fourati, Wided Souidene Mseddi*, Rabah Attia

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

17 Scopus citations

Abstract

In this paper, we propose an object detection methodology applied to Global Wheat Head Detection (GWHD) Dataset. We have been through two major architectures of object detection which are Faster R-CNN, and EfficientDet, in order to design a novel and robust wheat head detection model. We emphasize on optimizing the performance of our proposed final architectures. Furthermore, we have been through an extensive exploratory data analysis, data cleaning, data splitting and adapted best data augmentation techniques to our context. We use semi supervised learning, precisely pseudo-labeling, to boost previous supervised models of object detection. Moreover, we put much effort on ensemble learning including test time augmentation, multi-scale ensemble and bootstrap aggregating to achieve higher performance. Finally, we use weighted boxes fusion as our post processing technique to optimize our wheat head detection results. Our solution has been submitted to solve a research challenge launched on the GWHD Dataset which was led by nine research institutes from seven countries. Our proposed method was ranked within the top 6% in the above-mentioned challenge.

Original languageEnglish (US)
Pages (from-to)198-208
Number of pages11
JournalCanadian Journal of Remote Sensing
Volume47
Issue number2
DOIs
StatePublished - 2021

ASJC Scopus subject areas

  • General Earth and Planetary Sciences

Fingerprint

Dive into the research topics of 'Wheat Head Detection using Deep, Semi-Supervised and Ensemble Learning'. Together they form a unique fingerprint.

Cite this