Skip to content

SriRangaTarun/SIIM-Melanoma-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Melanoma Classification 🔬

A model based on EfficientNet-B7 to classify melanoma (a type of skin cancer) from skin images. In this project, I build a model based on EfficientNet-B7 to classify skin images as either being "Melanoma" or "Healthy". I use PyTorch XLA, EfficientNet PyTorch, and Albumentations to train the model using data from the SIIM-ISIC Melanoma Classification competition on Kaggle. This method achieves a ROC-AUC score of 0.845 on the public leaderboard which can definitely be taken closer to 0.9 with more tuning.

About

A model based on EfficientNet-B7 to classify melanoma (a type of skin cancer) from skin images.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published