Skip to content

Skin Cancer lesion classification using transfer learning

License

Notifications You must be signed in to change notification settings

gaurav8341/SkinCancerCapstone

Repository files navigation

SkinCancerCapstone

This project aims to create a classification model for the skin cancer lesion classification. The data used for this project is HAM10000 dataset which can be found here

In the Data Exploration notebook, the data is visualized and prepared for the actual model training like data splitting, upsampling.

In the Model Creation notebook, the transfer learning based approach is used to develop an actual model.

Required libraries

  • Pytorch
  • Numpy
  • pandas
  • matplotlib
  • Seaborn
  • OpenCV

About

Skin Cancer lesion classification using transfer learning

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published