Real-time Litter Recognition Using Improved YOLOv4 Tiny Algorithm
fully connected layer; litter; object detection; prediction time; YOLOv4-Tiny
Littered roads have become a familiar sight in India. The main reason is the increasing population and inefficient waste disposal system. Since garbage collectors cannot pick litter in all the places, there is a need for an efficient way to detect it. Hence, a machine learning-based object detection model is used. In this, we have applied an improved YOLOv4-Tiny algorithm to detect the garbage, classify it and make the detection process easier on custom datasets. We have improved the algorithm in terms of the object prediction time, this is done by replacing a max pooling layer with one of two layers present in a fully connected layer. When an input is given, the algorithm detects the litter in the image with a bounding box around it along with the label and confidence score. The proposed model reduces the prediction time by 0.517 milliseconds less than the original algorithm employed which concludes that the object is predicted faster. 2022 IEEE.
Shalini V.; Tangade S.; Prajna P.K.; Sangeetha J.P.; Azam F.; Anoop G.L.
MysuruCon 2022 - 2022 IEEE 2nd Mysore Sub Section International Conference
Institute of Electrical and Electronics Engineers Inc.
https://doi.org/10.1109/MysuruCon55714.2022.9972356 https://www.scopus.com/inward/record.uri?eid=2-s2.0-85145347420&doi=10.1109%2fMysuruCon55714.2022.9972356&partnerID=40&md5=770727b3bd0f9a0ea00f103ad5a04248
Shalini V., Reva University, School of Electronics and Commn. Engg, Bangalore, India; Tangade S., Christ (Deemed to Be University), Dept. of Cse, Bangalore, India; Prajna P.K., Reva University, School of Electronics and Commn. Engg, Bangalore, India; Sangeetha J.P., Reva University, School of Electronics and Commn. Engg, Bangalore, India; Azam F., School of Comp. Science & Engineering, Reva University, Bangalore, India; Anoop G.L., Christ (Deemed to Be University), Dept. of Cse, Bangalore, India