Skip to content

Skin cancer detection using deep learning model vgg-16 in pytorch. Gradcam allows to visualize discriminating features

Notifications You must be signed in to change notification settings

clemonster/Skin_Cancer_Detection_with_GradCam

 
 

Repository files navigation

MelaNet

Skin cancer classification

Robin Ali - Louise Badarani - Cyriac Parisot - Clément Ponsonnet - Ruoy Zhang

On average between 2 and 3 million skin cancers are diagnosed yearly world wide (World Health Organization). AI has been proven as a powerful diagnostic tool in medical fiels. We thus aim to develop a classifier to help dermathologist assess their diagnostics and understand the most prominent characteristics of each cancer types.

We trained and tested our models on the HAM10000 ("Human Against Machine with 10000 training images") dataset, a set of labeled dermatoscopic images from different populations, acquired and stored by different modalities.

7 types of skin cancers are being detected:

  • Actinic keratoses and intraepithelial carcinoma / Bowen's disease (akiec)
  • basal cell carcinoma (bcc)
  • benign keratosis-like lesions (solar lentigines / seborrheic keratoses and lichen-planus like keratoses, bkl)
  • dermatofibroma (df)
  • melanoma (mel)
  • melanocytic nevi (nv)
  • vascular lesions (angiomas, angiokeratomas, pyogenic granulomas and hemorrhage, vasc).

Our VGG16 model achieved 92% accuracy on our test set after 100 epochs on the classification task.

The GradCam was able to detect discriminating visual features of particular legion classes. See the images below for illustration:

The Original Picture:

Original Picture

The GradCam Overlay:

GradCam Overlay

The detailed presentation is in presentation.ipynb.

About

Skin cancer detection using deep learning model vgg-16 in pytorch. Gradcam allows to visualize discriminating features

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 96.5%
  • Shell 3.5%