Real-time attribute based deep learning network for traffic sign detection
Traffic sign detection is one of the key components of the Advanced Driving Assistant System (ADAS), which aims to detect and classify street signs in real time. However, traffic sign detection has challenges in real applications requiring high precision and real-time recall. Those challenges are du...
Saved in:
Main Authors: | , , , , |
---|---|
Format: | Conference or Workshop Item |
Published: |
2022
|
Subjects: | |
Online Access: | http://eprints.utm.my/id/eprint/98694/ http://dx.doi.org/10.1109/ICITEE56407.2022.9954110 |
Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
id |
my.utm.98694 |
---|---|
record_format |
eprints |
spelling |
my.utm.986942023-02-02T05:47:10Z http://eprints.utm.my/id/eprint/98694/ Real-time attribute based deep learning network for traffic sign detection Mohamed Elhawary, Hossamelden Suddamalla, Upendra Shapiai, Mohd. Ibrahim Wong, Anthony Zamzuri, Hairi T Technology (General) Traffic sign detection is one of the key components of the Advanced Driving Assistant System (ADAS), which aims to detect and classify street signs in real time. However, traffic sign detection has challenges in real applications requiring high precision and real-time recall. Those challenges are due to the small object size and class imbalance. Recently, researchers have proposed several techniques to improve the detection quality by enriching the features through a multiscale network, introducing attention mechanisms and augmentation techniques to improve the features of tiny objects. To overcome class imbalance researchers proposed cascaded networks and various loss functions. However, those existing techniques and mechanisms added more complexity to the model. Meanwhile, the imbalance affects single-stage networks such as YOLO, which causes a lower recall for minor classes. We proposed a new training method for a single-stage detection network, known as Real Time Attribute Based Deep Learning Detection Network (Real Time-Attribute DL). We introduced new attributes to the loss and Non-Maximum Suppression (NMS) to reduce the class number by categorizing it based on the shape of the traffic sign while maintaining the same number of classes. Our proposed method extends the YOLO detection head to have four main parameters: objectiveness, regression, class, and attribute. We modify the loss function to train the network jointly between class and attribute. We validate our proposed technique with Tsinghua-Tencent 100K(TT100K) as a benchmark dataset. The results show that our proposed technique improves the recall index from 85.85% to 94.26% in yolov4-tiny-31 with a 0.8% improvement in precision and improves the recall index from 93.51% to 96.68% in yolov4 with a drop by 2% in precision without adding extra complexity to the main network. The proposed technique offers a better recall index than the baseline, especially for imbalanced datasets such as TT100K datasets. 2022 Conference or Workshop Item PeerReviewed Mohamed Elhawary, Hossamelden and Suddamalla, Upendra and Shapiai, Mohd. Ibrahim and Wong, Anthony and Zamzuri, Hairi (2022) Real-time attribute based deep learning network for traffic sign detection. In: 14th International Conference on Information Technology and Electrical Engineering, ICITEE 2022, 18 - 19 October 2022, Yogyakarta, Indonesia. http://dx.doi.org/10.1109/ICITEE56407.2022.9954110 |
institution |
Universiti Teknologi Malaysia |
building |
UTM Library |
collection |
Institutional Repository |
continent |
Asia |
country |
Malaysia |
content_provider |
Universiti Teknologi Malaysia |
content_source |
UTM Institutional Repository |
url_provider |
http://eprints.utm.my/ |
topic |
T Technology (General) |
spellingShingle |
T Technology (General) Mohamed Elhawary, Hossamelden Suddamalla, Upendra Shapiai, Mohd. Ibrahim Wong, Anthony Zamzuri, Hairi Real-time attribute based deep learning network for traffic sign detection |
description |
Traffic sign detection is one of the key components of the Advanced Driving Assistant System (ADAS), which aims to detect and classify street signs in real time. However, traffic sign detection has challenges in real applications requiring high precision and real-time recall. Those challenges are due to the small object size and class imbalance. Recently, researchers have proposed several techniques to improve the detection quality by enriching the features through a multiscale network, introducing attention mechanisms and augmentation techniques to improve the features of tiny objects. To overcome class imbalance researchers proposed cascaded networks and various loss functions. However, those existing techniques and mechanisms added more complexity to the model. Meanwhile, the imbalance affects single-stage networks such as YOLO, which causes a lower recall for minor classes. We proposed a new training method for a single-stage detection network, known as Real Time Attribute Based Deep Learning Detection Network (Real Time-Attribute DL). We introduced new attributes to the loss and Non-Maximum Suppression (NMS) to reduce the class number by categorizing it based on the shape of the traffic sign while maintaining the same number of classes. Our proposed method extends the YOLO detection head to have four main parameters: objectiveness, regression, class, and attribute. We modify the loss function to train the network jointly between class and attribute. We validate our proposed technique with Tsinghua-Tencent 100K(TT100K) as a benchmark dataset. The results show that our proposed technique improves the recall index from 85.85% to 94.26% in yolov4-tiny-31 with a 0.8% improvement in precision and improves the recall index from 93.51% to 96.68% in yolov4 with a drop by 2% in precision without adding extra complexity to the main network. The proposed technique offers a better recall index than the baseline, especially for imbalanced datasets such as TT100K datasets. |
format |
Conference or Workshop Item |
author |
Mohamed Elhawary, Hossamelden Suddamalla, Upendra Shapiai, Mohd. Ibrahim Wong, Anthony Zamzuri, Hairi |
author_facet |
Mohamed Elhawary, Hossamelden Suddamalla, Upendra Shapiai, Mohd. Ibrahim Wong, Anthony Zamzuri, Hairi |
author_sort |
Mohamed Elhawary, Hossamelden |
title |
Real-time attribute based deep learning network for traffic sign detection |
title_short |
Real-time attribute based deep learning network for traffic sign detection |
title_full |
Real-time attribute based deep learning network for traffic sign detection |
title_fullStr |
Real-time attribute based deep learning network for traffic sign detection |
title_full_unstemmed |
Real-time attribute based deep learning network for traffic sign detection |
title_sort |
real-time attribute based deep learning network for traffic sign detection |
publishDate |
2022 |
url |
http://eprints.utm.my/id/eprint/98694/ http://dx.doi.org/10.1109/ICITEE56407.2022.9954110 |
_version_ |
1758578007056842752 |
score |
13.211869 |