Bounding Box CutMix와 표준화 거리 기반의 IoU를 통한 재활용품 탐지
Recyclable Objects Detection via Bounding Box CutMix and Standardized Distance-based IoU
0
조회
0
다운로드
1
인용
0
참고문헌
키워드
초록
In this paper, we developed a deep learning-based recyclable object detection model. The model is developed based on YOLOv5 that is a one-stage detector. The deep learning model detects and classifies the recyclable object into 7 categories: paper, carton, can, glass, pet, plastic, and vinyl. We propose two methods for recyclable object detection models to solve problems during training. Bounding Box CutMix solved the no-objects training images problem of Mosaic, a data augmentation used in YOLOv5. Standardized Distance-based IoU replaced DIoU using a normalization factor that is not affected by the center point distance of the bounding boxes. The recyclable object detection model showed a final mAP performance of 0.91978 with Bounding Box CutMix and 0.91149 with Standardized Distance-based IoU.
Abstract
In this paper, we developed a deep learning-based recyclable object detection model. The model is developed based on YOLOv5 that is a one-stage detector. The deep learning model detects and classifies the recyclable object into 7 categories: paper, carton, can, glass, pet, plastic, and vinyl. We propose two methods for recyclable object detection models to solve problems during training. Bounding Box CutMix solved the no-objects training images problem of Mosaic, a data augmentation used in YOLOv5. Standardized Distance-based IoU replaced DIoU using a normalization factor that is not affected by the center point distance of the bounding boxes. The recyclable object detection model showed a final mAP performance of 0.91978 with Bounding Box CutMix and 0.91149 with Standardized Distance-based IoU.
참고문헌 (18)
- 김필, "하이브리드 거버넌스 분석: 재활용 폐기물 수거 거부 사례를 중심으로", 한국거버넌스학회보, 27(2), pp.29~56 (2020), 한국거버넌스학회[DOI]
- J. Redmon, "You only look once: Unified, Real-time Object Detection", In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.779~788 (2016)
- S. Ren, ""Faster R-CNN : Towards Real-time Object Detection with Region Proposal Networks", Advances in Neural Information Processing Systems, 28 (2015)
- K. He, "Mask R-CNN", Proceedings of the IEEE International Conference on Computer Vision, pp.2961~2969 (2017)
- "YOLOv5"[URL]
- A. Bochkovskiy, "Yolov4: Optimal Speed and Accuracy of Object Detection", arXiv preprint, arXiv:2004.10934, ~ (2020)
- H. J. Lee, "Implementation of Recyclable Trash Object Detection Model Based on YOLOv5", Proceedings of the Institute of Electronics and Information Engineers, pp.1066~1069 (2021)
- T. Devries, "Improved Regularization of Convolutional Neural Networks with Cutout", arXiv preprint, arXiv:1708.04552, ~ (2017)
- G. Kang, "Patchshuffle Regularization", arXiv preprint, arXiv:1707.07103, ~ (2017)
- H. Zhang, "Mixup: Beyond Empirical risk Minimization", arXiv preprint, arXiv : 1710. 09412, ~ (2017)
- S. Yun, "Cutmix: Regularization Strategy to Train Strong Classifiers with Localizable Features", Proceedings of the IEEE, 2019, 6023~6032
- G. Ghiasi, "Simple Copy-paste is a Strong Data Augmentation Method for Instance Segmentation", Proceedings of the IEEE, 2021, 2918~2928
- N. Bodla, "Improving Object Detection with one line of code", arXiv preprint, arXiv:1704.04503, ~ (2017)
- H. Rezatofighi, "Generalized Intersection Over Union: A Metric and A loss for Bounding box Regression", Proceedings of the IEEE, 2019, 658~666
- Z. Zheng, "Distance-iou Loss: Faster and Better Learning for Bounding box Regression", Proceedings of the AAAI Conference on Artificial Intelligence, 34(07), pp.12993~13000 (2020)
- "2020 AI Grand Challenge"[URL]
- M. Everingham, "The Pascal Visual Object Classes (voc)Challenge", International Journal of Computer Vision, 88(2), pp.303~338 (2010)
- "YOLOv5 pre-trained weights of COCO dat aset"[URL]