Influence of Training Parameters on Real-Time Similar Object Detection Using YOLOv5s

被引:3
作者
Kvietkauskas, Tautvydas [1 ]
Stefanovic, Pavel [2 ]
机构
[1] Vilnius Gediminas Tech Univ, Dept Informat Technol, LT-10223 Vilnius, Lithuania
[2] Vilnius Gediminas Tech Univ, Dept Informat Syst, LT-10223 Vilnius, Lithuania
来源
APPLIED SCIENCES-BASEL | 2023年 / 13卷 / 06期
关键词
YOLOv5s; real-time object detection; construction details dataset; similar objects;
D O I
10.3390/app13063761
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Object detection is one of the most popular areas today. The new models of object detection are created continuously and applied in various fields that help to modernize the old solutions in practice. In this manuscript, the focus has been on investigating the influence of training parameters on similar object detection: image resolution, batch size, iteration number, and color of images. The results of the model have been applied in real-time object detection using mobile devices. The new construction detail dataset has been collected and used in experimental investigation. The models have been evaluated by two measures: the accuracy of each prepared model has been measured; results of real-time object detection on testing data, where the recognition ratio has been calculated. The highest influence on the accuracy of the created models has the iteration number chosen in the training process and the resolution of the images. The higher the resolution of the images that have been selected, the lower the accuracy that has been obtained. The small iteration number leads to the model not being well trained and the accuracy of the models being very low. Slightly better results were obtained when the color images were used.
引用
收藏
页数:15
相关论文
共 30 条
[1]   Adaptive Drone Identification and Neutralization Scheme for Real-Time Military Tactical Operations [J].
Ajakwe, Simeon Okechukwu ;
Ihekoronye, Vivian Ukamaka ;
Akter, Rubina ;
Kim, Dong-Seong ;
Lee, Jae Min .
36TH INTERNATIONAL CONFERENCE ON INFORMATION NETWORKING (ICOIN 2022), 2022, :380-384
[2]  
Alkentar S.M., 2021, J. Eng, V27, P19, DOI [10.31026/j.eng.2021.08.02, DOI 10.31026/J.ENG.2021.08.02]
[3]   K-Means clustering and neural network for object detecting and identifying abnormality of brain tumor [J].
Arunkumar, N. ;
Mohammed, Mazin Abed ;
Abd Ghani, Mohd Khanapi ;
Ibrahim, Dheyaa Ahmed ;
Abdulhay, Enas ;
Ramirez-Gonzalez, Gustavo ;
de Albuquerque, Victor Hugo C. .
SOFT COMPUTING, 2019, 23 (19) :9083-9096
[4]  
Bochkovskiy A., 2020, YOLOv4: Optimal Speed and Accuracy of Object Detection, Vabs/2004.10934, P1
[5]  
Bozaba E., 2021, P 2021 29 SIGNAL PRO, P1
[6]   Investigation of YOLOv5 Efficiency in iPhone Supported Systems [J].
Dluznevskij, Daniel ;
Stefanovic, Pavel ;
Ramanauskaite, Simona .
BALTIC JOURNAL OF MODERN COMPUTING, 2021, 9 (03) :333-344
[7]   Deep Multi-Modal Object Detection and Semantic Segmentation for Autonomous Driving: Datasets, Methods, and Challenges [J].
Feng, Di ;
Haase-Schutz, Christian ;
Rosenbaum, Lars ;
Hertlein, Heinz ;
Glaser, Claudius ;
Timm, Fabian ;
Wiesbeck, Werner ;
Dietmayer, Klaus .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2021, 22 (03) :1341-1360
[8]  
Guohua Zhu, 2020, 2020 International Conference on Artificial Intelligence in Information and Communication (ICAIIC), P070, DOI 10.1109/ICAIIC48513.2020.9065216
[9]  
Han GX, 2022, AAAI CONF ARTIF INTE, P780
[10]   Residual dense collaborative network for salient object detection [J].
Han, Yibo ;
Wang, Liejun ;
Cheng, Shuli ;
Li, Yongming ;
Du, Anyu .
IET IMAGE PROCESSING, 2023, 17 (02) :492-504