pith. sign in

arxiv: 2307.04537 · v1 · pith:K7XDLB2Enew · submitted 2023-07-10 · 💻 cs.CV · cs.AI

Q-YOLOP: Quantization-aware You Only Look Once for Panoptic Driving Perception

classification 💻 cs.CV cs.AI
keywords modelbdd100kdrivingquantization-awaresegmentationtrainingdatasetsdetection
0
0 comments X
read the original abstract

In this work, we present an efficient and quantization-aware panoptic driving perception model (Q- YOLOP) for object detection, drivable area segmentation, and lane line segmentation, in the context of autonomous driving. Our model employs the Efficient Layer Aggregation Network (ELAN) as its backbone and task-specific heads for each task. We employ a four-stage training process that includes pretraining on the BDD100K dataset, finetuning on both the BDD100K and iVS datasets, and quantization-aware training (QAT) on BDD100K. During the training process, we use powerful data augmentation techniques, such as random perspective and mosaic, and train the model on a combination of the BDD100K and iVS datasets. Both strategies enhance the model's generalization capabilities. The proposed model achieves state-of-the-art performance with an mAP@0.5 of 0.622 for object detection and an mIoU of 0.612 for segmentation, while maintaining low computational and memory requirements.

This paper has not been read by Pith yet.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.