Small datasets for fruit detection with transfer learning

Dan Dai, Junfeng Gao, Simon Parsons, Elizabeth Sklar

Research output: Chapter in Book/Published conference outputConference publication

Abstract

A common approach to the problem of fruit detectionin images is to design a deep learning network andtrain a model to locate objects, using bounding boxes to identifyregions containing fruit. However, this requires sufficient dataand presents challenges for small datasets. Transfer learning,which acquires knowledge from a source domain and brings thatto a new target domain, can produce improved performance inthe target domain. The work discussed in this paper shows theapplication of transfer learning for fruit detection with smalldatasets and presents analysis between the number of trainingimages in source and target domains. This investigation is basedon three datasets: two containing tomatoes and one containingstrawberries. Experimental results indicate that transfer learningcan enhance prediction with limited data.
Original languageEnglish
Title of host publication4th UK-RAS Conference for PhD and Early Career Researchers Proceedings
Number of pages2
DOIs
Publication statusPublished - 8 Jul 2021

Fingerprint

Dive into the research topics of 'Small datasets for fruit detection with transfer learning'. Together they form a unique fingerprint.

Cite this