DermaAI - Deep learning based Skin lesions Classification
CNNs based classification of skin lesions
Github page - https://github.com/Amritpal-001/DermaAI
Aim
deploying deep learning models for Derma diagnosis
Goals -
- Train models
- Test models
- Python based GUI framework to deploy model
- Prediction probabilities
- Choosing model function - Multiple models for different diseases
- Image augmentation before making prediction
- Model optimisation
- Model Pruining
- GradCAM for explainability of model
| Roadmap |
|---|
 |
Frameworks used -
- Model training - Tensorflow, Pytorch
| Data |
 |  |  |
Updates -
- 13 june 2021 - Github updated and cleaned. Next step is to add documentation to the previous code first.