Skip to content

htchu/DeepTongueVis

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 

Repository files navigation

System architecture:

alt text

Design steps:

  1. Transfer learning for tongue fissures: an ImageNet-pretrained ResNet50 model is used to perform a transfer learning for images with tongue fissures and images without tongue fissures. That is, only the last one layer is replaced with a binary classifier and retrained. Other layers are intact without any modification.
  2. Tongue fissure visualization: apply grad-CAM to the trained model to visualize tongue fissures. An example is here: https://github.com/htchu/DeepTongueVis/blob/master/gradcam_sample_code.py

Sample results:

alt text

About

Deep Learning studies for Tongue Images

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages