A progressive weighted average weight optimisation ensemble technique for fruit and vegetable classification

Document Type

Conference Proceeding

Publication Title

2020 16th IEEE International Conference on Control, Automation, Robotics and Vision, (ICARCV)

First Page

303

Last Page

308

Publisher

IEEE

School

School of Engineering

RAS ID

32761

Funders

Edith Cowan University / Natural Sciences and Engineering Research Council of Canada

Comments

Hameed, K., Chai, D., & Rassau, A. (2020, December). A progressive weighted average weight optimisation ensemble technique for fruit and vegetable classification [Paper presentation]. 2020 16th International Conference on Control, Automation, Robotics and Vision (ICARCV), Shenzhen, China. https://doi.org/10.1109/ICARCV50220.2020.9305474

Abstract

© 2020 IEEE. Image classification of fruit and vegetables at supermarket self-checkouts is a complex problem. Significant variations in the size, shape and colour of objects are involved, along with potentially large variations in the environmental conditions, must be accommodated to implement such a robust and effective system. Convolution Neural Networks (CNNs) have shown promising results for object classifications. However, the scarcity of training datasets due to the diversity of varieties and applications of fruit and vegetable classification is a significant limitation to the CNN implementation for this task. To overcome this, we propose the use of transfer learning and ensemble technique. Specifically, a transfer learning based weighted average weight optimisation ensemble technique is applied to the weights of GoogleNet and MobileNet by transfer learning the pre-trained CNNs using a custom dataset. Two hyperparameter optimisation techniques have been applied in a sequential way to identify the effective weights progressively. The optimised weights are used as an input to a normalised exponential softmax layer to estimate the final probability distribution for classification. A comparative evaluation among standalone GoogleNet, MobileNet and different levels of ensemble has been presented, which supports the adoption of this technique as a solution in a real-world supermarket environment.

DOI

10.1109/ICARCV50220.2020.9305474

Access Rights

subscription content

Link to Full Text

Share

 
COinS