/WBC-AI-Imageing

2023 graduation project - White Blood Cell(WBC) detection with AI

Primary LanguageJupyter Notebook

WBC-AI-Imageing

2023 Sookmyung W. Univ. graduation project
Project: White Blood Cell(WBC) detection with AI

This project aims to detect four types of white blood cells: eosinophils, lymphocytes, macrophages, and neutrophils, through object detection. Bounding boxes are drawn using OpenCV and objects are cropped from them to serve as input for a classification task performed by a model. We employed a MobileNetV2-based model trained on each cell image.

Table of Contents

Background

Respiratory diseases pose a significant threat to human life, with high incidence rates in modern society. Among various respiratory disease tests, bronchoalveolar lavage fluid examination is the most fundamental, requiring discriminating and counting immunocytes on slide images, which is a labor-intensive task. Additionally, differential blood cell counts play an important role in general health checkups.

The current method for differential blood cell counts involves manual identification and counting of cells, resulting in long processing times and frequent errors due to reduced accuracy and consistency. As a result, recent research has focused on image preprocessing and deep learning-based methods for differential blood cell counts. However, previous research has limitations such as incomplete representation of immune cell features, low cell detection rates, and low classification accuracy, despite reducing processing time compared to manual methods. Therefore, this project aims to leverage image preprocessing and deep learning techniques to enhance detection performance.

Usage

Image Preprocessing Section → Modeling

Section Explanation

Image Preprocessing for annotating bounding-box

bounding-box annotation result bounding-box annotation result

Clustering

clustering result of test image
In the actual project, clustering was performed on approximately 50,000 cropped cell images obtained from the original dataset.

label 0

label 1

label 2

label 3

label 4

label 0

Modeling

The final model used was 'MobileNetV2'. During the model selection process, LeNet-5 and VGG16 were also considered. The input image was reshaped to '96x96x3'.

Contributors

기지원 조유림 윤혜경