A comparative study of YOLOv5 models performance for image localization and classification

被引:0
作者
Horvat, Marko [1 ]
Jelecevic, Ljudevit [1 ]
Gledec, Gordan [1 ]
机构
[1] Univ Zagreb, Dept Appl Comp, Fac Elect Engn & Comp, Unska 3, HR-10000 Zagreb, Croatia
来源
CENTRAL EUROPEAN CONFERENCE ON INFORMATION AND INTELLIGENT SYSTEMS, CECIIS 2022 | 2022年
关键词
computer vision; image classification; deep learning; deep convolutional neural networks; YOLO; OBJECT DETECTION; TIME;
D O I
暂无
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
YOLOv5 is one of the latest and often used versions of a very popular deep learning neural network used for various machine learning tasks, mainly in computer vision. The YOLO algorithm has steadily gained acceptance in the data science community due to its superior performance in complex and noisy data environments, availability, and ease of use in combination with widely used programming languages such as Python. This paper aims to compare different versions of the YOLOv5 model using an everyday image dataset and to provide researchers with precise suggestions for selecting the optimal model for a given problem type. The obtained results and the implemented YOLOv5 models are available for non-commercial use at: https://github.com/mhorvat/YOLOv5-models-comparison
引用
收藏
页码:349 / 356
页数:8
相关论文
共 37 条
  • [1] Bochkovskiy A., 2020, YOLOv4: Optimal Speed and Accuracy of Object Detection, DOI DOI 10.48550/ARXIV.2004.10934
  • [2] Chiang D., 2020, AIZOOTech
  • [3] Doljanin D., 2021, P 2021 44 INT CONVEN, P1690, DOI [10.23919/MIPRO52101.2021.9596869, DOI 10.23919/MIPRO52101.2021.9596869]
  • [4] Pavement distress detection and classification based on YOLO network
    Du, Yuchuan
    Pan, Ning
    Xu, Zihao
    Deng, Fuwen
    Shen, Yu
    Kang, Hua
    [J]. INTERNATIONAL JOURNAL OF PAVEMENT ENGINEERING, 2021, 22 (13) : 1659 - 1672
  • [5] Tinier-YOLO: A Real-Time Object Detection Method for Constrained Environments
    Fang, Wei
    Wang, Lin
    Ren, Peiming
    [J]. IEEE ACCESS, 2020, 8 : 1935 - 1944
  • [6] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [7] Rich feature hierarchies for accurate object detection and semantic segmentation
    Girshick, Ross
    Donahue, Jeff
    Darrell, Trevor
    Malik, Jitendra
    [J]. 2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, : 580 - 587
  • [8] github, 2022, YOLOv6: a single -stage object detection framework dedicated to industrial applications
  • [9] Guanhao Yang, 2020, 2020 IEEE 6th International Conference on Computer and Communications (ICCC), P1398, DOI 10.1109/ICCC51575.2020.9345042
  • [10] Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2015, 37 (09) : 1904 - 1916