Subway driver emotion identification method and device based on deep learning

Abstract

The invention provides a subway driver emotion identification method and device based on deep learning. The method comprises the following steps: obtaining a driver emotional state database, includingface images and corresponding emotional states; constructing a deep learning network model used for emotional state identification; using the driver emotional state database to train the deep learning network model, and solidifying the structure and parameters of the trained deep learning network model to obtain an emotional state identification model; and collecting a real-time working video ofa subway driver, extracting multiple continuous frames of images according to a preset frame rate, and inputting the multiple continuous frames of images to the emotional state identification model toobtain the emotional state identification result of the subway driver. The emotional state of a subway driver is monitored in real time by using a deep learning network model based on a 3D convolution neural network and a stacked auto-encoder. The identification precision is high. The abnormal working state of a driver can be found early, and the safety of driving can be ensured.

Description

technical field [0001] The present invention relates to the field of artificial intelligence, and more specifically, to a method and device for identifying emotions of subway drivers based on deep learning. Background technique [0002] In recent years, the average running speed and traffic density of urban rail transit trains have been continuously increasing. Although the train driving safety equipment and the driver's standardized operation level have been greatly improved, the driver's inattention and fatigue driving may occur during the value ride process. The probability of serious threats to the safe operation of trains such as loss of control ability caused by accidental injuries and diseases still exists. [0003] At present, the existing train safety equipment of my country's urban rail transit basically does not have the function of driver's vigilance and control. It does not recognize and alarm the emotional state of the driver's personal work, and the system fun...

AI Technical Summary

Problems solved by technology

[0003] At present, the existing train driving safety equipment of my country's urban rail transit basically does not have the function of driver's vigilance and control. It does not recognize and alarm the emotional state of the driver's personal work, and the system function cannot meet the actual operation requirements.

Images