/SIIM-ISIC-Melanoma-Classification

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

SIIM-ISIC-Melanoma-Classification

Melanoma Classification Competition

DATA SETUP : Assumes that you have kaggleAPI installed

mkdir ./JPEG Melanoma 256x256
mkdir ./JPEG Melanoma 2019 256x256
kaggle datasets download -d cdeotte/jpeg-melanoma-256x256
kaggle datasets download -d cdeotte/jpeg-isic2019-256x256
unzip -q jpeg-melanoma-jpeg-melanoma-256x256.zip -d JPEG Melanoma 256x256
unzip -q jpeg-isic2019-jpeg-isic2019-256x256.zip -d JPEG Melanoma 2019 256x256

Usage

# Train Model:
python src/train.py

You can run the notebook to showcase the training process.

OverView of the Model Architecture :