CentroidNet: A Deep Neural Network for Joint Object Localization and Counting

Dijkstra, K., van de Loosdrecht, J., Schomaker, L. R. B. & Wiering, M. A., 2019, ECML PKDD 2018: Machine Learning and Knowledge Discovery in Databases. Brefeld, U., Curry, E., Daly, E., MacNamee, B., Marascu, A., Pinelli, F., Berlingerio, M. & Hurly, N. (eds.). Cham: Springer, p. 585-601 17 p. ( Lecture Notes in Computer Science; vol. 11053).

Research output: Chapter in Book/Report/Conference proceedingChapterAcademicpeer-review

In precision agriculture, counting and precise localization of crops is important for optimizing crop yield. In this paper CentroidNet is introduced which is a Fully Convolutional Neural Network (FCNN) architecture specifically designed for object localization and counting. A field of vectors pointing to the nearest object centroid is trained and combined with a learned segmentation map to produce accurate object centroids by majority voting. This is tested on a crop dataset made using a UAV (drone) and on a cell-nuclei dataset which was provided by a Kaggle challenge. We define the mean Average F1 score (mAF1) for measuring the trade-off between precision and recall. CentroidNet is compared to the state-of-the-art networks YOLOv2 and RetinaNet, which share similar properties. The results show that CentroidNet obtains the best F1 score. We also explicitly show that CentroidNet can seamlessly switch between patches of images and full-resolution images without the need for retraining.
Original languageEnglish
Title of host publicationECML PKDD 2018
Subtitle of host publicationMachine Learning and Knowledge Discovery in Databases
EditorsUlf Brefeld, Edward Curry, Elizabeth Daly, Brian MacNamee, Alice Marascu, Fabio Pinelli, Michele Berlingerio, Neil Hurly
Place of PublicationCham
Number of pages17
ISBN (Electronic)978-3-030-10997-4
ISBN (Print)978-3-030-10996-7
Publication statusPublished - 2019

Publication series

Name Lecture Notes in Computer Science
ISSN (Print)0302-9743

ID: 76676094