Method and apparatus for fatigue prediction based on analogue brain wave data

An electronic device, comprising:

• a processor;
• and
• a memory, used for storing a program of a method for fatigue prediction based on analogue brain wave data;
• after the electronic device is powered on and the processor runs the program of the method for fatigue prediction based on analogue brain wave data, the electronic device performs a method for fatigue prediction based on analogue brain wave data;
• the method comprises:
• extracting an eye video sequence when the testee is awake and corresponding brain wave data in an awake state, an eye video sequence when the testee is fatigued and corresponding brain wave data in a fatigue state as training sample data;
• training a preset generative adversarial network, and adjusting parameters to obtain a target generative adversarial network;
• wherein the target generative adversarial network includes a generator model, an evaluator model, an adversarial discriminator and the fatigue discriminator;
• obtaining the fatigue discriminator from the target adversarial network model;
• collecting an eye video sequence based on a video capture device;
• inputting the eye video sequence into a default fatigue discriminator to obtain predicted analogue brain wave data;
• and
• outputting the analogue brain wave data to the fatigue discriminator to discriminate a fatigue state.