Loading repository data…
jayavanth18 / repository
Deep Learning-based Skin Cancer Classification using FastAI and ResNet | Includes dataset, Jupyter Notebook, PR & ROC curves, confusion matrix, and literature review.
A lightweight end-to-end deep learning pipeline for classifying dermatoscopic skin lesion images using transfer learning with FastAI and PyTorch.
This project implements a Convolutional Neural Network (CNN) using FastAI and a PyTorch backend to classify skin cancer lesions from dermatoscopic images.
It uses a ResNet18 model pre-trained on ImageNet and fine-tunes it on a curated subset of the Skin Cancer ISIC dataset.
⚠️ Disclaimer: This project is for educational & research purposes only. It is not a substitute for professional medical diagnosis.
⚠️ Note: Due to resource and training time constraints, the notebook uses only a subset of classes (some classes were removed) from the original dataset.
| Feature | Description |
|---|---|
| 🧠 Transfer Learning | Pre-trained ResNet18 model fine-tuned with FastAI best practices. |
| ⚡ Simplified Workflow | FastAI abstracts away boilerplate PyTorch code for faster prototyping. |
| 🔄 Data Augmentation | Uses aug_transforms (flips, rotation, zoom, warp) for generalization. |
| 📊 Comprehensive Evaluation | Includes Confusion Matrix, Precision-Recall Curve, and ROC Curve. |
| 🎯 Prediction Demo | Load the trained model and run single-image predictions with probabilities. |
| 🧪 Test-Time Augmentation (TTA) | Achieved 80.68% accuracy by averaging predictions over augmented images. |
| 📈 Learning Rate Finder | Automatically selects the best learning rate with learn.lr_find(). |
Skin-Cancer-Detection-CNN-FastAI/
├── code.ipynb # Main Jupyter Notebook
├── README.md # Project documentation
├── literature_review.pdf # Background research
We use the publicly available Skin Cancer ISIC dataset from Kaggle: 🔗 Skin Cancer 9 Classes (ISIC) - Kaggle
Important Note: To reduce training time and manage resources, the notebook does not use all 9 classes. One or more classes were intentionally excluded during training. You can modify the notebook to include all classes if needed.
fine_tune + fit_one_cycle# 1. Clone the repository
git clone https://github.com/YOUR_USERNAME/Skin-Cancer-Detection-CNN-FastAI.git
cd Skin-Cancer-Detection-CNN-FastAI
# 2. Install dependencies
pip install fastai timm matplotlib scikit-learn
# 3. Download dataset (manually or via Kaggle CLI)
# Example using Kaggle CLI:
kaggle datasets download -d nodoubttome/skin-cancer9-classesisic
unzip skin-cancer9-classesisic.zip
# 4. Launch Jupyter
jupyter notebook code.ipynb
The model’s performance was assessed using multiple metrics.
Confusion Matrix, Optimal Learning Rate & Top Losses
Precision-Recall & ROC Curves
This project is licensed under the MIT License. See the LICENSE file for details.
⭐️ If you found this repository helpful, please consider giving it a star!
Selected from shared topics, language and repository description—not editorial ratings.
Pranav-Uniyal /
A deep learning–based skin disease classification system using ResNet152V2 with a Streamlit web app and a Jupyter notebook for model training and testing.
thesangeeth /
A deep learning-based image classification model to detect and classify various skin disorders using convolutional neural networks (CNNs). Built with Python, TensorFlow, and Jupyter Notebook.
Developed a deep learning-based skin cancer classification system using CNN models to identify melanoma and other skin lesions from dermoscopic images. Duration: 6 months. Domain: Healthcare AI. Tech Stack: Python, TensorFlow/Keras, OpenCV, NumPy, Pandas.achieving high classification accuracy using image datasets and Jupyter Notebook tools
