akarataev / gleam-model

Gleam model for VK Hackathon 2018

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

icon
Gleam model

Skin cancer screening nerual network model



About Gleam

Skin cancer is in third place in terms of the incidence of oncology detection in Russian men and second in women. The five-year threshold is experienced by only half of the patients. Even with the second stage of melanoma and with proper treatment. Therefore, an early production diagnosis is practically the only way to get a favorable prognosis for treatment. The solution could be a mobile application, with which you can make an early diagnosis and promptly seek treatment.

Dataset

To train this model the data to use is a set of images from the International Skin Imaging Collaboration: Mellanoma Project ISIC.

Preprocessing

The following preprocessing tasks are developed for each image:

  1. Visual inspection to detect images with low quality
  2. Image resizing: Transform images to 255x256x3
  3. Crop images

Final result

AUC/ROC curve chart VGG16

For transfer learning model:

Training and validation loss VGG16

For transfer learning (30 epochs):

For fine tuning (10 epochs):

Training and validation accuracy VGG16

For transfer learning (30 epochs):

For fine tuning (10 epochs):

About

Gleam model for VK Hackathon 2018


Languages

Language:Python 100.0%