Skin Lesion Detection System

Ensemble-based Skin Lesion Detection System

Overview

This repository contains the code and resources for an Ensemble-based Skin Lesion Detection System designed to assist in the identification and classification of skin lesions through image analysis. This system leverages advanced machine learning techniques to provide accurate and timely diagnostics.

If you use our code, please use it after installation through the .yaml file we uploaded

System Architecture

The system is composed of the following key modules:

Data Exchange Module

• Image Receiver: Receives images for analysis.
• Diagnostic Result Sender: Sends the diagnosis results to the user.

Diagnostic Result Analysis Module

• Grad-CAM Visualization: Provides a heatmap visualization for identifying the focus areas of the model.
• Lesion Segmentation: Segments the lesion from the skin image.

Skin Lesion Inference Module

• Lesion Detection: Detects the presence of a lesion in the skin image.
• Lesion Classification: Classifies the type of lesion (e.g., atopic dermatitis, insect bite, psoriasis).
• Ensemble Decision: Uses soft voting to combine the results of multiple models to improve accuracy.

Getting Started

Environment Setup

Using Docker

We have transitioned to using Docker for environment setup to ensure consistency across all setups. To set up your environment using Docker, follow these steps:

1. Build the Docker image from the provided Dockerfile: docker build -t lesion .
2. Run the Docker container: docker run -it --rm --gpus all -p 8000:8000 lesion

Using Conda (Alternative)

For users who prefer setting up their environment using Conda, please note that while the project has primarily moved to a Docker-based setup for its environment, it's still possible to use Conda based on earlier project configurations.

If you specifically need or prefer to use Conda for your environment setup, refer to the project state at commit 54ac160 for the environment.yaml file that was used before the transition to Docker. This file contains the necessary configuration to set up your environment using Conda.

To revert to or manually adjust your environment.yaml based on the specifications found in commit 54ac160, you may follow these steps:

1. Navigate to the project's Git repository.
2. Use the following Git command to checkout the environment.yaml file from commit 54ac160:

   git checkout 54ac160 -- environment.yaml

   This command will check out the environment.yaml file from commit 54ac160 into your current working directory. Please be aware that this action might overwrite any local changes you've made to the environment.yaml file. Ensure to back up your current environment file if necessary.
3. Once you have the environment.yaml file from commit 54ac160, create the Conda environment with the command:

   conda env create --file environment.yaml

Demo instruction using pre-trained model

• Currently, this repository does not provide a pre-trained model for public use due to size constraints and licensing issues. If you are interested in a demonstration of the Ensemble-based Skin Lesion Detection System using a pre-trained model, please follow these steps:

1. Obtain a pre-trained model by contacting the repository maintainers at [email protected] or [email protected].
2. Place the pre-trained model in the designated directory within the project.

• Run with pretrained model

1. Download pretrained models 
2. Move pre-trained efficientnet_models model into `classification/efficientnet_models`
3. Move pre-trained Mask R-CNN and YOLOv8 into `segmentation/mrcnn_models and segmentation/yolo_models`
5. python server/manage.py runserver [your ip:port]

Dataset Information and Usage

The datasets used in this Ensemble-based Skin Lesion Detection System can be accessed through the following AI Hub URLs:

• AI Hub Dataset 1(피부 질환 진단 의료 이미지
• AI Hub Dataset 2(소아청소년 피부질환 이미지 데이터)

To access and use these datasets, please ensure that you follow the rules and regulations provided by AI Hub. This may include procedures for registration, data handling, privacy compliance, and licensing agreements.

Note: The datasets provided by AI Hub are subject to their terms of use, and it is the responsibility of the users to adhere to these terms. If you have any questions about the usage of the datasets, please refer to the AI Hub's guidelines or contact their support for assistance.

Sample Results

The images below demonstrate the sample results from the ATOMOM Lesion Analyzer for both desktop and mobile interfaces.

Desktop Interface Result

Mobile Interface Result

Note: If the images are not displaying, please ensure that the links are correct and that the GitHub repository settings allow for image hosting.

Acknowledgements

This implementation has been based on these repository yolov8, Mask R-CNN, EfficientNet.