/Melanoma-Image-Classification

Developing a Melanoma Detector with Neural Networks and Flask for Deployment

Primary LanguageJupyter NotebookMIT LicenseMIT

Melanoma Image Classification with Flask Application

Project Page

Please navigate to this page for the original final report for the project:

https://datascisteven.github.io/Melanoma-Image-Classification

Presentation Link: https://prezi.com/view/JoLKnGuw0ZFBYFZra0c3/

Flask application

While I had been exploring implementation through Flutter for app deployment, Flask seemed much more feasible given my time constraints and level of expertise.

Home page:

The homepage asks for the user to upload a JPEG of any size into the application and to press SUBMIT once done.

Results page:

Upon pressing SUBMIT, you automatically get transferred to the Results page, and you are given a message to get the mole checked out or that it is just another beauty mark. The confidence level of that prediction is also given.

Folder Structure:

├── README.md                   <- the top-level README for reviewers of this project
├── _notebooks					<- folder containing all the project notebooks
│   ├── albumentation.ipynb		<- notebook for displaying augmentations
│   ├── EDA.ipynb				<- notebook for dataset understanding and EDA
│   ├── folders.ipynb			<- notebook for image folder management
│   ├── holdout.ipynb			<- notebook for predicting on holdout sets
│   ├── preaugmentation.ipynb	<- notebook for models with imbalanced dataset
│   ├── postaugmentation.ipynb	<- notebook for models with dataset post-augmentations
│   ├── pretrained.ipynb		<- notebook for pretrained models
│   └── utils.py  				<- py file with self-defined functions
├── final_notebook.ipynb        <- final notebook for capstone project
├── _data                       <- folder of csv files (csv)
├── MVP Presentation.pdf		<- pdf of the MVP presentation
├── _Melanoma-Flask				<- folder with Flask application
└── utils.py					<- py file with self-defined functions

Contact Information:

Steven Yan

Email: stevenyan@uchicago.edu

LinkedIn: https://www.linkedin.com/in/datascisteven

Github: https://www.github.com/datascisteven

References:

International Skin Imaging Collaboration. SIIM-ISIC 2020 Challenge Dataset. International Skin Imaging Collaboration https://doi.org/10.34970/2020-ds01 (2020).

Rotemberg, V. et al. A patient-centric dataset of images and metadata for identifying melanomas using clinical context. Sci. Data 8: 34 (2021). https://doi.org/10.1038/s41597-021-00815-z

ISIC 2019 data is provided courtesy of the following sources:

Tschandl, P. et al. The HAM10000 dataset, a large collection of multi-source dermatoscopic images of common pigmented skin lesions. Sci. Data 5: 180161 doi: 10.1038/sdata.2018.161 (2018)

Codella, N. et al. “Skin Lesion Analysis Toward Melanoma Detection: A Challenge at the 2017 International Symposium on Biomedical Imaging (ISBI), Hosted by the International Skin Imaging Collaboration (ISIC)”, 2017; arXiv:1710.05006.

Marc Combalia, Noel C. F. Codella, Veronica Rotemberg, Brian Helba, Veronica Vilaplana, Ofer Reiter, Allan C. Halpern, Susana Puig, Josep Malvehy: “BCN20000: Dermoscopic Lesions in the Wild”, 2019; arXiv:1908.02288.

Codella, N. et al. “Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)”, 2018; https://arxiv.org/abs/1902.03368