Skip to content

grewe/StrokeChangeMLTraining

Repository files navigation

StrokeChangeMLTraining

StrokeChange is a proof-of-concept system that uses computer vision and machine learning to monitor stroke patients in-situ towards the goal of monitoring patient progress and changes. This innovative system can detect changes in a patient's facial expressions, which may indicate changes in their condition, such as the onset of new symptoms or the worsening of existing ones. Various approaches including detection, classification and regression to solve this problem are implemented and compared. A dataset was curated for training and testing of StrokeChange. This work used a modified version of the dataset originally found at Facial_Droop_and_Facial_Paralysis_image, https://www.kaggle.com/datasets/kaitavmehta/facial-droop-and-facial-paralysis-image

This is the repository for the python Tensorflow based Colabs for training the 3 implementations of StrokeChange: Detection Only, Region Detect to Classification and Region Detect to Regression

image

Directories

LabeledData-8-classes

colab to train the 8-class StrokeChange Detector Only Implementation

LabeledData-6-classes

colab to train the 6-class StrokeChange Detector Only Implementation

LabeledData-Eye-Mouth

colab to train the StrokeChange Region Detection to Classification & Regression implementations' Eye&Mouth Detector

LabeledData-Mouth-Regression

colab to train the StrokeChange Region Detection to Regression implementation's MOUTH Regression model

LabeledData-Eye-Regression

colab to train the StrokeChange Region Detection to Regression implementation's EYE Regression model

LabeledData-Mouth-Classifier-4Classes

colab to train the StrokeChange Region Detection to Classification implementation's MOUTH Classifier

LabeledData-Eye-Classifier-4Classes

colab to train the StrokeChange Region Detection to Classification implementation's EYE Classifier

GeneralColabs

base colabs/notebooks used to train detectors, classifiers and regression models used in StrokeChange ALSO see other directories for specific directories for spefic colabs used in actual training that are based off of these general colabs

ColabsForProcessingData

contains colabs/notebooks for processing data to prepare it for training

About

This is the repository for the python Tensorflow based Colabs for training the 3 implementations of StrokeChange: Detection Only, Region Detect to Classification and Region Detect to Regression

Resources

License

Stars

1 star

Watchers

1 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors