All experiments were performed on a GeForce RTX 2070 GPU with 8GB memory using the Pytorch framework. The implementation is available at: https://github.com/p2irc/UDA4POC
Open resource ↗p2irc/UDA4POC · lines:81-104Paper record
Unsupervised Domain Adaptation For Plant Organ Counting
arXiv · 2 Sept 2020 · 10.48550/arxiv.2009.01081
Abstract
Supervised learning is often used to count objects in images, but for counting small, densely located objects, the required image annotations are burdensome to collect. Counting plant organs for image-based plant phenotyping falls within this category. Object counting in plant images is further challenged by having plant image datasets with significant domain shift due to different experimental conditions, e.g. applying an annotated dataset of indoor plant images for use on outdoor images, or on a different plant species. In this paper, we propose a domain-adversarial learning approach for domain adaptation of density map estimation for the purposes of object counting. The approach does not assume perfectly aligned distributions between the source and target datasets, which makes it more broadly applicable within general object counting and plant organ counting tasks. Evaluation on two diverse object counting tasks (wheat spikelets, leaves) demonstrates consistent performance on the target datasets across different classes of domain shift: from indoor-to-outdoor images and from species-to-species adaptation.
Code and data availability
The paper provides two paper-specific public assets: the authors' implementation code for the domain-adversarial counting model on GitHub, and the authors' newly created GWHD dot annotations deposited on figshare. Other URLs are cited prior-work datasets, not paper-specific assets.
To evaluate our method, we created dot annotations for 67 images from the GWHD which are used as ground truth. These annotations are made publicly available at https://doi.org/10.6084/m9.figshare.12652973.v2 .
Open resource ↗10.6084/m9.figshare.12652973.v2 · lines:105-155