Skip to content
/ ECG Public
forked from CoolDude937/ECG

Emotional Cardiography (ECG)

Notifications You must be signed in to change notification settings

Hasan3773/ECG

 
 

Repository files navigation

Emotional Cardiography (ECG)

Created by Mark Do, Hasan Tahir, Andy Ma, Mckale Chung

Inspiration

The inspiration for ECG comes form the tremendous labor shortage Canada is currently having in the medical industry and the desire to make safety and security accessible to everyone whos staying in medical facilities.

What it does

ECG uses a ML model we trained and python OpenCV to be able to detect a humans emotions based of facial expressions. Then using a heart rate sensor hooked up to an Arduino to simulate a professional grade cardiac event monitor, we take in a patients bpm using C++. Then it sends both pieces of data to a web application made in JavaScript in real time using various libraries and techniques. Finally, the web application made using react creates a friendly user interface for the doctor to be able to see the probability of the patients possible emotions, the bpm and a suggestions section which gives the doctor recommendations based off our data.

How we built it

First we created a sequential model using 8 convolutional layers based on the fer2013 data set obtained from Kaggle. fer2013

We chose a 90/10 split on the dataset, resulting in around 24,000 training images and 2400 validation images. Our model has 4 convolutional block followed by two 64 units dense layer, and finally a softmax output. The general architecture of our convolution network uses (3x3) convolutional kernel with ELU activation followed by a batch normalization layer. A block consists of two of these kernel-activation-batchnorm components, followed by dropout and max pooling. The model has 4.7 million parameters and was trained over 75 epochs, resulting in a validation accuracy of 75%.

We used OpenCV and the HaasCaade face clasifier to detect faces, which is then passed into our own model to output emotion predictions.

Then after some optimization, we sent the model to our website by using the external flask library to return the live footage into an html file.

On the hardware side, we used an KY-039 heartbeat sensor and an Arduino to simulate a professional cardiogram. We developed an algorithm to approximate the average BPM of the last 4 seconds and used the serial monitor to send the information to our Express backend.

Challenges we ran into

One of the main challenges we ran into was the ability to transfer data in real time, whether that be from python to JavaScript, sending a live streamed from python to html, or sending C++ readings to JavaScript. Most of our time ended up being spent debugging various different transferring techniques we found. Another challenged we had was our model program was inconsistent with successfully initializing, one attempt it would work and the next after no syntax changes it would fail. Another issue we ran into in the OpenCV side was getting all the versions to line up in the virtual environment we created, eventually after enough pip installs we found out about poetry and used that to standardize all of our version control.

Challenges of getting BPM: The first challenge was the lack of an actual pulse rate sensor. The heartbeat sensor we had instead does not return a BPM but a number from 0 to 1023 when blood moves through the finger and the sensor picks up a change in light. We had to figure out an algorithm that would convert this to BPM. The second challenge was to use an algorithm that correctly approximated a BPM in short intervals, as we needed speed for the demo. We eventually decided on 4 seconds, which was long enough to give decent readings.

Accomplishments that we're proud of

BPM accomplishments: -syncing a buzzer beep with the heartbeat -being able to find a BPM without a pulse rate sensor

What we learned

-Proper equipment or potentially more useful sensors can make things a lot easier

What's next for Emotional-Cardiography (ECG)

-Perhaps switching to a pulse rate sensor, and tuning a bit to better approximate the average BPM at the same rate or faster.

About

Emotional Cardiography (ECG)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 96.7%
  • JavaScript 1.7%
  • Python 0.7%
  • C++ 0.4%
  • SCSS 0.2%
  • HTML 0.2%
  • CSS 0.1%