Skip to content

Latest commit

 

History

History
15 lines (11 loc) · 652 Bytes

README.md

File metadata and controls

15 lines (11 loc) · 652 Bytes

Skin Lesion Classification

Check it from here: Classifier

About the data ..."

The dataset used here is PAD-UFES-20

About the network ...

I have used here a fine-tuned ResNet18 netowrk

Run the code

First downlowd the data from Kaggle

Then clone this code using git clone https://github.com/mohamedelhacen/skin-cancer-classification

Run the following command to prepare the data, train and test the network, and save checkpoints pyton main.py --train.epochs 100