Fine-Tuning Swin Transformer for Helmet Detection: A Comparative Study with CNNs
摘要
In this research, we analyse helmet detection with transformer-based and convolutional methods. First, we used a CNN-based classifier for helmet classification and YOLOv8 for detection. We then used a refined Swin Transformer to expand our investigation, assessing both models on inference time, precision, recall and F1-score. According to our comparative analysis, transformer-based models can maintain competitive inference efficiency while outperforming CNNs in accuracy.