Skip to content

A medical project aims to use Machine Learning model in identifing and detecting change of patient's wound size, border and colour.

Notifications You must be signed in to change notification settings

lucas-project/Woud_Detection_Machine_Learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

97 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Version_1.py is a methods of implementing border detection and colour detection. It used Tensorflow as framework, over 300 artifact and real wound images as training materials to create a model that is able to identify wound's border and its change.

#Version_1.py Version_1 used TensorFlow with Kera, U-net architecture with VGG16 encoder

To run the code: Install required libraries first: ''' pip install numpy opencv-python tensorflow requests pillow scikit-learn scikit-image matplotlib ''' Then: ''' python version_1.py '''

'version_1.py': Main program 'v1_border': functions related to border identification 'v1_colour': functions related to colour handling 'v1_evaluation': functions realted to evaluate training performance

'json_images' folder: images with masking in JSON format 'json_masking' folder: JSON format masking 'jj_images' folder: images with masking in .jpg format 'jj_masking' folder: .jpg format masking 'evaluation' folder: images for evaluating model's performance

'fake_jj' folder: JSON format masking 'fake_wound' folder: fake images with .jpg format maksing 'fake_evaluation' folder: fake evaluation images

'wound_segmentation_model.h5': the model file

pip install opencv-contrib-python

All training materials need to add mask manually using Labelbox masks

The mask that is ready for use mask-2

Wound image, model's identification result, and the result after fine tune wound-result-fine tune result

Evaluation dataset samples evaluation dataset

different type of training materials used different trained artifact wound images

Training and validation loss curve training and validation loss curve

In case of complicated wound, doctor have the option to circle out wound's border manually, blue line is the border marked manually by a doctor, and the black and white images show the identification result by our algorithms. manual circle out complicated wound

About

A medical project aims to use Machine Learning model in identifing and detecting change of patient's wound size, border and colour.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages