Skip to content

LidaDavydova/MelNevClassification

Repository files navigation

MelNevClassification

RU версия

Описание

MelNevClassification - проект, нацеленный на решение проблемы распознавания (классифицирования) раковых новообразований на теле, в частности, меланом и родинок с помощью анализа дермотологическихснимков сверточной нейронной сетью. Мы выделили именно эти 2 класса (меланома и родинка), чтобы посмотреть на динамику обучения нейронной сети.

Дермотологические снимки для обучения и тестирования были взяты из официальных открытых источников.

Протестируйте сами тут

Или запуском программы для тестирования - запустить файл testModel.ipynb

Источники данных

HAM10000

ISIC Challenge:

ISIC Challenge 2024

ISIC Challenge 2020

ISIC Challenge 2019

Детали реализации

Обработка данных

Из выбранных источников, описанных выше, были взяты только снимки меланом и родинок. Вышло следующее: classes

Чтобы увеличить количество изображений меланом, мы попробовали убрать волосы на изображениях кожи, где они были, тем самым увеличили датасет на 800 фоток.

Пример:

hair_remove_example

Затем появилась идея попробовать менять цвет кожи, создавая свои маски для некоторых снимков.

Пример:

mask_example1 mask_example2

Для обучения классы изображений должны быть +- одинакого распределены, поэтому мы собрали датасет из следующего количества:

data_distrib

Обучение нейронной сети

В файле MelNev_train.ipynb показано обучение модели model-72.keras.

Ее архитектура:

История обучения:

Показания тестирования обученной нейронной сети:

72% точность

EN version

Description

MelNevClassification is a project aimed at solving the problem of recognizing (classifying) cancerous neoplasms on the body, in particular melanomas and moles, using a convolutional neural network to analyze dermatological images. We identified these two classes (melanoma and mole) to look at the dynamics of neural network training.

Dermatological images for training and testing were taken from official open sources.

Test it yourself here

Or by running the testing program - run the testModel.ipynb file

Data sources

HAM10000

ISIC Challenge:

ISIC Challenge 2024

ISIC Challenge 2020

ISIC Challenge 2019

Implementation details

Data processing

From the selected sources described above, only images of melanomas and moles were taken. The following came out: classes

To increase the number of melanoma images, we tried to remove hair from skin images where it was, thereby increasing the dataset by 800 photos.

Example:

hair_remove_example

Then an idea came up to try changing the skin color by creating our own masks for some images.

Example:

mask_example1 mask_example2

For training, the image classes should be +- equally distributed, so we collected a dataset from the following amount:

data_distrib

Implementation details

Data processing

From the selected sources described above, only images of melanomas and moles were taken. The following came out: classes

To increase the number of melanoma images, we tried to remove hair from skin images where it was, thereby increasing the dataset by 800 photos.

Example:

hair_remove_example

Then an idea came up to try changing the skin color by creating our own masks for some images.

Example:

mask_example1 mask_example2

For training, the image classes should be +- equally distributed, so we collected a dataset from the following amount:

data_distrib

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors