/Skin-cancer-detection

This project has been built with Flask which use trained weights for skin classification problem. I trained model with pytorch and got 85 AUC score.

Primary LanguageJupyter Notebook

Skincancer-detection

data contain "model_0.bin" which is trained model to classify melanoma cancer.

and se_resnext weights are pretrained model for resnet and imagenet is used.

this model achieved 85 AUC score

To run this web app:

python api.py

or

python3 api.py