Skip to content

This is a production grade project which is used to classify eight different types of Skin Diseases using CNN. It is trained on top of VGG16 architecture and used MLFLOW for model versioning. It also uses DVC pipeline to automate the complete each stage of the project. Finally I have dockerized the project and Hosted In AWS using EC2 instance, ECR.

License

Notifications You must be signed in to change notification settings

bharath2468/Skin-Disease-Classification

Repository files navigation

End-to-End-Skin-Disease-Classification-using-MLflow-DVC

Workflows

  1. Update config.yaml
  2. Update secrets.yaml [Optional]
  3. Update params.yaml
  4. Update the entity
  5. Update the configuration manager in src config
  6. Update the components
  7. Update the pipeline
  8. Update the main.py
  9. Update the dvc.yaml

MLflow

cmd
  • mlflow ui

dagshub

dagshub

MLFLOW_TRACKING_URI=
MLFLOW_TRACKING_USERNAME=
MLFLOW_TRACKING_PASSWORD=
python script.py

Run this to export as env variables:

export MLFLOW_TRACKING_URI="FILL YOUR DETAILS"

export MLFLOW_TRACKING_USERNAME="FILL YOUR DETAILS" 

export MLFLOW_TRACKING_PASSWORD="FILL YOUR DETAILS"

DVC cmd

  1. dvc init
  2. dvc repro
  3. dvc dag

About MLflow & DVC

MLflow

  • Its Production Grade
  • Trace all of your expriements
  • Logging & taging your model

DVC

  • Its very lite weight for POC only
  • lite weight expriements tracker
  • It can perform Orchestration (Creating Pipelines)

AWS-CICD-Deployment-with-Github-Actions

1. Login to AWS console.

2. Create IAM user for deployment

#with specific access

1. EC2 access : It is virtual machine

2. ECR: Elastic Container registry to save your docker image in aws


#Description: About the deployment

1. Build docker image of the source code

2. Push your docker image to ECR

3. Launch Your EC2 

4. Pull Your image from ECR in EC2

5. Lauch your docker image in EC2

#Policy:

1. AmazonEC2ContainerRegistryFullAccess

2. AmazonEC2FullAccess

3. Create ECR repo to store/save docker image

- Save the URI: 

4. Create EC2 machine (Ubuntu)

5. Open EC2 and Install docker in EC2 Machine:

#optinal

sudo apt-get update -y

sudo apt-get upgrade

#required

curl -fsSL https://get.docker.com -o get-docker.sh

sudo sh get-docker.sh

sudo usermod -aG docker ubuntu

newgrp docker

6. Configure EC2 as self-hosted runner:

setting>actions>runner>new self hosted runner> choose os> then run command one by one

7. Setup github secrets:

AWS_ACCESS_KEY_ID=

AWS_SECRET_ACCESS_KEY=

AWS_REGION = us-east-1

AWS_ECR_LOGIN_URI = demo>>  566373416292.dkr.ecr.ap-south-1.amazonaws.com

ECR_REPOSITORY_NAME = skin_disease_classification

About

This is a production grade project which is used to classify eight different types of Skin Diseases using CNN. It is trained on top of VGG16 architecture and used MLFLOW for model versioning. It also uses DVC pipeline to automate the complete each stage of the project. Finally I have dockerized the project and Hosted In AWS using EC2 instance, ECR.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published