Screening X-ray baggage is crucial for border security, but image quality and available prohibited data are challenges. Lack of data in training AI models can affect the detection results. Here, we evaluate object detection based on YOLO-v8 with an imbalanced dataset from SIXray dataset. The proposed X-ray baggage security screening framework consists of image acquisition, annotation, classification, and threat detection phases. The YOLO-v8 architecture is used to train the model, allowing for quick and accurate identification of potential threats. Evaluation methods such as confusion matrix and precision-recall curve provide a comprehensive view of the classifier’s performance. The model was trained using 17,498 images and 1642 images for validation. The model performance achieves the largest recall value of 92%, and the largest precision of 93.80%. However, even with an F1 score is 63.81%, the experiment showing the YOLO-v8 algorithm can detect small objects and overlap objects in different orientations. This shows that the Yolo architecture can handle imbalanced datasets.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Evaluation Imbalanced Dataset for X-Ray Baggage Security Screening Based on YOLO-v8 for Enhanced Threat Detection

  • Mohd Shafry Mohd Rahim,
  • Ajune Wanis Ismail,
  • Devi Willieam Anggara,
  • Nurfathin Atika Nawawi,
  • Mohd Shahrizal Sunar,
  • Farhan Mohammed

摘要

Screening X-ray baggage is crucial for border security, but image quality and available prohibited data are challenges. Lack of data in training AI models can affect the detection results. Here, we evaluate object detection based on YOLO-v8 with an imbalanced dataset from SIXray dataset. The proposed X-ray baggage security screening framework consists of image acquisition, annotation, classification, and threat detection phases. The YOLO-v8 architecture is used to train the model, allowing for quick and accurate identification of potential threats. Evaluation methods such as confusion matrix and precision-recall curve provide a comprehensive view of the classifier’s performance. The model was trained using 17,498 images and 1642 images for validation. The model performance achieves the largest recall value of 92%, and the largest precision of 93.80%. However, even with an F1 score is 63.81%, the experiment showing the YOLO-v8 algorithm can detect small objects and overlap objects in different orientations. This shows that the Yolo architecture can handle imbalanced datasets.