Machine Learning and Knowledge Extraction (Dec 2022)

Synthetic Data Generation for Visual Detection of Flattened PET Bottles

  • Vitālijs Feščenko,
  • Jānis Ārents,
  • Roberts Kadiķis

DOI
https://doi.org/10.3390/make5010002
Journal volume & issue
Vol. 5, no. 1
pp. 14 – 28

Abstract

Read online

Polyethylene terephthalate (PET) bottle recycling is a highly automated task; however, manual quality control is required due to inefficiencies of the process. In this paper, we explore automation of the quality control sub-task, namely visual bottle detection, using convolutional neural network (CNN)-based methods and synthetic generation of labelled training data. We propose a synthetic generation pipeline tailored for transparent and crushed PET bottle detection; however, it can also be applied to undeformed bottles if the viewpoint is set from above. We conduct various experiments on CNNs to compare the quality of real and synthetic data, show that synthetic data can reduce the amount of real data required and experiment with the combination of both datasets in multiple ways to obtain the best performance.

Keywords