A lie detector based on Pytorch for a 2022 R&E school project.
I sometime play the lie game with my brother and sister. This equipment looks like real lie detector. But the fundamental law is simple. I'd like to improve this lie detector using Deep learning Technology.
I set up presumption that a lie relates with our face and our heart rate. Because when I lie, my heart beats fast and my facial expression is changed.
I tracked face expression with Mediapipe FaceMesh model and heart rate with Arduino Sensor.
I made the UI for data collecting using Flask webpage. This webpage is quite simple but essential for training. Using this UI, I get the image and heart rate numeric figure during lying and not lying.
MODEL V1
I made simple Linear Net with Pytorch. I will experiment many type of models to improve the performance.
MODEL V1-2
The performance of this model didn't improve because of some reasons.
The factors are like this.
- We don't reflect facial expression position data.
- We don't preprocess landmark position.
Solution I think.
- BatchNormalization
- Dropout Network
I changed some features of the network. And train again it.
v1 Model Result
MODEL V2
I made Neural Network with Convolutional Network becauase linear net doesn't reflect the facial expresssion emotion. But this network doesn't working well too...