ISSN: 1820-0206

First published in 1950

 

 

 

About the journal

Editorial board

All issues

Latest issue

Issue in progress

Publishing Policy

Guide for authors

Reviwers guidelines

Submit your article

Contact

 

 

 

doi: 10.5937/str2600007Z

Vol. 75,   No.2 (2025),   Pages: 46-54

Impressum

Contents

 

Research of the Impact of the Number of Epochs and Dataset Size on the Performance of the YOLO Algorithm


Snežana Zurovac
Jelena Jevremović
Ninko Miletić
Vasilija Joksimović

 

                   

This study investigates the influence of dataset size and the number of training epochs on the performance of YOLOv8 object detection models. Three model variants: YOLOv8s, YOLOv8m, and YOLOv8xl, were trained using five datasets of increasing size (200 to 1000 images) and five epoch configurations ranging from 10 to 50 epochs. Model performance was evaluated using standard validation metrics, including mean Average Precision at an IoU threshold of 0.5-0.95 (mAP50-95), precision, and recall. The results indicate that increasing dataset size and training duration generally improves detection performance, with performance gains diminishing beyond approximately 40 training epochs. Model generalization capability was further assessed using a separate test set consisting of 32 manually annotated images, employing a similarity coefficient that integrates localization accuracy, classification correctness, and detection completeness. Additional analyses of true positives (TP), false positives (FP), and false negatives (FN) were conducted at confidence thresholds of 0.25, 0.50, and 0.85 to examine detection robustness under varying confidence constraints. Among the evaluated models, YOLOv8m provided the best balance between detection accuracy and computational efficiency. YOLOv8s, although less accurate, remained well suited for real-time inference and deployment on resource-constrained systems. In contrast, YOLOv8x did not exhibit consistent performance improvements relative to its increased computational complexity.


Key words: YOLOv8, Object detection, Dataset size, Training epochs, Model performance.

 

DOWNLOAD PDF

 

 

Copyright ©, Vojnotehnički institut, 2007.  - www.vti.mod.gov.rs