top of page
2023年第一届研究、创新、创意展
(R.I.C.E'23)
Project ID:
ITCS101 (Virtual Evaluation)
X-Ray Baggage Object Detection Using Neural Network for Safety Purpose
Project Title:
Category:
Information Technology/ Computer Science
Inventors:
Samuel Ato Gyasi Otabir, Associate Professor EUR ING. Ir Ts Dr Lim Wei Hong, Assistant Professor Ir Ts Dr Tiang Sew Sun, Dr Leong HungYang, Associate Professor EUR ING. Ir Ts Dr
Institution/Company:
UCSI University
Invention Description/ Abstract:
Airport security a matter of urgent attention and this calls for important measures to ensure that all baggage that travel within the airport contain non-harmful objects that may put people at the airport in danger. Over the last decade, x-yay machines have evolved to scan baggage whilst an airport officer verifies that the content of the baggage bag contain benign items; if anomaly items are identified the owner of the baggage is called aside to have further enquiries. However, this process is time consuming usually takes about 5-15s and moreover lacks accuracy at times due to fatigue on the side of the officer checking the scanned x-ray images. This paper is prepared to develop a convolutional neural network (CNN) to aid in the process of X-Ray baggage object detection. A proposed model, YOLOv5 is trained with a 47,677-x-ray image dataset known as PIDray, which contains 12 threat object classes: gun, bullet, knife, wrench, pliers, power bank, baton, lighter, hammer, and handcuffs. The model is trained on Google Colab whereas the deployment of the model is done on PyCharm Community. The proposed model is evaluated based on its precision, recall, f1-score, and mean average precision (mAP) collectively known as performance metrics. A comparison between the proposed model, YOLOv5 and another similar model, YOLOv3 is performed under the basis of their performance metrics. The mAP of the proposed model, YOLOv5 is 90.0%, the precision is seen to be 90.4%, recall of 89.6%, and a f1-score of 87.4%. The proposed model when deployed in real-time has an inference time of 0.008s with a minimum accuracy of 83.0%. The proposed model also exhibits good objectness, classification, and intersection over union (IoU). Therefore, the model is applicable for x-ray object detection to increase safety at the airport.
Invention Technical Description
The most paramount step before training any convolutional neural network is the acquisition of a dataset. A dataset may consist of images which belong to several classes of interest. The dataset used for this project is a custom dataset with 12 classes of unwanted objects in a baggage, the dataset is known as Prohibited Item Detection Dataset (PIDray). For dataset annotation, a platform known as RoboFlow is used to annotate the images. Since the PIDray had 12 classes of object, every class was differentiated by a set of colors; red for the gun class, blue for the bullet class, yellow for the knife class, aqua for the pliers class, light brown for the power bank class, violet for the baton class, green for the lighter class, dark green for the sprayer class, sky blue for the hammer class, dark brown for the handcuffs class and a lighter shade of green for scissors. The annotations were saved in the YOLO format and exported to Google Colab for model training. The PIDray dataset which consists of 47, 677 images are used for training the models. All the images in the dataset are resized to 416 × 416 as an image pre-processing step to help improve the performance metrics of the module. The development of the neural network is firstly trained with optimized hyperparameters and an incremental epoch of 50, 100, 150, and 200. The second part of the development phase is deploying the trained neural network locally. The performance metrics used in evaluating the model performance involves the precision, recall, f1-score, mAP@0.5, and mAP@0.5:0.95. Google Colab is used for the training phase of the model and for the deployment process is performed by using PyCharm, command prompt and the local computer camera.Upon completion of detailed comparison between YOLOv3 and YOLOv5, it is safe and sound to conclude that YOLOv5 is the best fit for x-ray baggage object detection using neural networks for safety purposes. This is a reason being that YOLOv5 has the highest recorded performance metrics (precision, recall, f1-score, mAP@0.5, and mAP@0.5:0.95): 90.4%, 84.6%, 87.4%, 90% and 66.7% respectively. Furthermore, the testing phase of the model proofs to be flawless: the proposed model can detect accurately all objects of all the 12 classes in the PIDray dataset and detect objects when they are clustered on top of each other. To conclude, the YOLOv5 model was deployed locally on a computer to run real time, and the inference speed was recorded to be ±97.8ms.
Demostration/ Presentation Video
Poster/ Broucher/ Invention Photo
Video Link
Poster Link
bottom of page