Open AccessCGBNet: A Deep Learning Framework for Compost Classification
Author(s) -
Suchisrit Gangopadhyay,
Anthony Zhai
Publication year - 2022
Publication title -
ieee access
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.587
H-Index - 127
ISSN - 2169-3536
DOI - 10.1109/access.2022.3201099
Subject(s) - aerospace , bioengineering , communication, networking and broadcast technologies , components, circuits, devices and systems , computing and processing , engineered materials, dielectrics and plasmas , engineering profession , fields, waves and electromagnetics , general topics for engineers , geoscience , nuclear engineering , photonics and electrooptics , power, energy and industry applications , robotics and control systems , signal processing and analysis , transportation
Minimizing waste entering US landfills has become increasingly difficult. Composting organic waste can assist in reducing the effects of landfill pollution and greenhouse gas emissions. To help automate this process, we apply computer vision techniques to train a classifier to distinguish between green and brown compost. The lack of labeled compost data and the challenges and dangers posed by collecting such data create a problem for the automation of composting. In this study, we propose CGBNet, a deep learning framework to differentiate between green and brown compost for the creation of healthy compost piles with an effective ratio of carbon to nitrogen. First, superclass and subclass classification are proposed as two methods of labeling and classifying data. Six deep learning models are then trained and evaluated on a newly formed dataset of 1,960 images using both label structures. Next, transfer learning is applied for feature extraction to boost performance using state-of-the-art deep learning models. The models showed improvements in performance generalization with less overfitting. The best-performing model, CGBNet, used transfer learning and subclass classification to obtain an accuracy of 95%, a 16% improvement in accuracy from the best performing superclass classification model trained without transfer learning. Superclass classification also showed significant improvements in performance and practicality with transfer learning, allowing for robust compost recognition systems to be more extensible while requiring less data and fewer parameters to train.