This paper describes an integrated approach to fatigue classification that combines a face detector based on the multi-task convolutional neural network (MTCNN) algorithm with a binary image classification model. An MTCNN algorithm that extracts facial regions from input images is developed, pre-processes these images, and augments them. A convolutional neural network composed of convolutional and fully connected layers will analyze facial features to discriminate between “active” and “fatigue” states. The architecture proposed demonstrates robustness through data augmentation and dropout regularization. An Adam optimizer and a binary cross-entropy loss, designed in the training strategy, have been utilized. In the performance evaluation, all the above-mentioned assessment metrics are used, including training accuracy curves, confusion matrices, receiver operator characteristics analysis, and precision-recall. The effectiveness of the architecture is validated on a dataset of active and fatigue states, showing its capability for real-world applications.

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

Multi-task Cascaded Convolutional Neural Network Foundation for Driver Drowsiness Detection

  • Venkata Sudiksha Madeti,
  • V. K. Harini,
  • Manas Ranjan Prusty

摘要

This paper describes an integrated approach to fatigue classification that combines a face detector based on the multi-task convolutional neural network (MTCNN) algorithm with a binary image classification model. An MTCNN algorithm that extracts facial regions from input images is developed, pre-processes these images, and augments them. A convolutional neural network composed of convolutional and fully connected layers will analyze facial features to discriminate between “active” and “fatigue” states. The architecture proposed demonstrates robustness through data augmentation and dropout regularization. An Adam optimizer and a binary cross-entropy loss, designed in the training strategy, have been utilized. In the performance evaluation, all the above-mentioned assessment metrics are used, including training accuracy curves, confusion matrices, receiver operator characteristics analysis, and precision-recall. The effectiveness of the architecture is validated on a dataset of active and fatigue states, showing its capability for real-world applications.