Split-Attention 백본 네트워크를 활용한 차선 인식에 관한 연구
A Study on Lane Detection Based on Split-Attention Backbone Network
0
조회
0
다운로드
0
인용
0
참고문헌
키워드
초록
본 논문에서는 split-attention 네트워크를 백본으로 특징을 추출하는 차선인식 CNN 네트워크를 제안한다. split-attention은 CNN의 특징 추출 과정에서 feature map의 각 channel에 가중치를 부여하는 방법으로, 빠르게 변화하는 자동차의 주행 환경에서 안정적으로 이미지의 특징을추출할 수 있다. Tusimple 데이터 셋을 활용하여 본 논문에서 제안하는 네트워크를 학습·평가하였으며, 백본 네트워크의 레이어 수에 따른 성능 변화를 비교·분석 하였다. 평가 결과 최대96.26%의 정확도로 최신 연구에 준하는 결과를 얻었으며, FP의 경우 0.0234(2.34%)로 비교 연구 중 가장 좋은 결과를 보여준다. 따라서, 실제 차량의 주행 환경 등에서도 본 연구에서 제안하는 모델을 사용하여 오인식 없이 안정적인 차선 인식이 가능하다.
Abstract
This paper proposes a lane recognition CNN network using split-attention network as a backbone to extract feature. Split-attention is a method of assigning weight to each channel of a feature map in the CNN feature extraction process; it can reliably extract the features of an image during the rapidly changing driving environment of a vehicle. The proposed deep neural networks in this paper were trained and evaluated using the Tusimple data set. The change in performance according to the number of layers of the backbone network was compared and analyzed. A result comparable to the latest research was obtained with an accuracy of up to 96.26, and FN showed the best result. Therefore, even in the driving environment of an actual vehicle, stable lane recognition is possible without misrecognition using the model proposed in this study.
참고문헌 (23)
- Cai Z., "Cascade r-cnn: Delving into high quality object detection", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.6154~6162 (2018)
- Chen K., "Hybrid task cascade for instance segmentation", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.4974~4983 (2019)
- Chen P., "Efficient Road Lane Marking Detection with Deep Learning", 2018 IEEE 23rd International Conference on Digital Signal Processing(DSP), pp.1~5 (2018)
- Chen Z., "PointLaneNet: Efficient end-to-end CNNs for Accurate Real-Time Lane Detection", 2019 IEEE Intelligent Vehicles Symposium(IV), pp.2563~2568 (2019)
- He K., "Mask r-cnn", Proceedings of the IEEE International Conference on Computer Vision, pp.2961~2969 (2017)
- He K., "Deep residual learning for image recognition", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.770~778 (2016)
- He T., "Bag of tricks for image classification with convolutional neural networks", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.558~567 (2019)
- Hochreiter S., "The vanishing gradient problem during learning recurrent neural nets and problem solutions", International Journal of Uncertainty, Fuzziness and Knowledge-Based Systems, 6(2), pp.107~116 (1998)
- Hou Y., "Learning lightweight lane detection cnns by self attention distillation", Proceedings of the IEEE International Conference on Computer Vision, pp.1013~1021 (2019)
- Hu J., "Squeeze-and-excitation networks", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.7132~7141 (2018)
- Kingma D. P., "Adam: A method for stochastic optimization", arXiv preprint arXiv:1412.6980, ~ (2014)
- Ko Y., "Key Points Estimation and Point Instance Segmentation Approach for Lane Detection", arXiv preprint arXiv:2002.06604, ~ (2020)
- Krizhevsky A., "Advances in Neural Information Processing Systems", pp.1097~1105 (2012)
- Li X., "Selective kernel networks", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.510~519 (2019)
- Neven D., "Towards End-to-End Lane Detection: An Instance Segmentation Approach", 2018 IEEE Intelligent Vehicles Symposium(IV), pp.286~291 (2018)
- Parashar A., "Scnn : An accelerator for compressed-sparse convolutional neural networks", ACM SIGARCH Computer Architecture News, 45(2), pp.27~40 (2017)
- Ren S., "Advances in Neural Information Processing Systems", pp.91~99 (2015)
- Ruder S., "An overview of gradient descent optimization algorithms", arXiv preprint arXiv:1609.04747, ~ (2016)
- Szegedy C., "Going deeper with convolutions", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.1~9 (2015)
- "Tusimple" (2017)[URL]
- Wang Z., "Lanenet: Real-time lane detection networks for autonomous driving", arXiv preprint arXiv:1807.01726, ~ (2018)
- Yoo H., "Gradient-Enhancing Conversion for Illumination-Robust Lane Detection", IEEE Transactions on Intelligent Transportation Systems, 14(3), pp.1083~1094 (2013)[DOI]
- Zhang H., "Resnest: Split-attention networks", arXiv preprint arXiv:2004.08955, ~ (2020)
