Skip to content
This repository has been archived by the owner on Jul 28, 2024. It is now read-only.
/ cellseg Public archive

Code for the paper titled "Advancing instance segmentation and WBC classification in peripheral blood smear through domain adaptation: A study on PBC and the novel RV-PBS datasets" published on Elsevier's Expert Systems With Applications (ESWA) journal.

Notifications You must be signed in to change notification settings

Jimut123/cellseg

Repository files navigation

Advancing instance segmentation and WBC classification in peripheral blood smear through domain adaptation: A study on PBC and the novel RV-PBS datasets


Abstract

Automating blood cell counting and detection from smear slides holds significant potential for aiding doctors in disease diagnosis through blood tests. However, existing literature has not adequately addressed using whole slide data in this context. This study introduces the novel RV-PBS dataset, comprising ten distinct peripheral blood smear classes, each featuring multiple multi-class White Blood Cells per slide, specifically designed, for instance segmentation benchmarks. While conventional instance segmentation models like Mask R-CNN exhibit promising results in segmenting medical artifact instances, they face challenges in scenarios with limited samples and class imbalances within the dataset. This challenge prompted us to explore innovative techniques such as domain adaptation using a similar dataset to enhance the classification accuracy of Mask R-CNN, a novel approach in the domain of medical image analysis. Our study has successfully established a comprehensive pipeline capable of segmenting, detecting, and classifying blood samples from slides, striking an optimal balance between computational complexity and accurate classification of medical artifacts. This advancement enables precise cell counting and classification, facilitating doctors in refining their diagnostic analyses.

This is the part of my Master's thesis where we segmented white blood cell via Mask RCNN (Aniket's part) and used Domain Adaptation to detect the cells.

pip install --upgrade --no-cache-dir gdown

Datasets used


The dataset is annotated using CVAT. We are planning to release an extended version of this dataset soon. If you are a haematologist, then you could help us by annotating and adding more data. Please make sure that the data is ethically cleared before uploading new data in public servers, such as Github.

Snapshot of dataset creation using CVAT

Some relevant stuffs from the paper

Please study the paper for getting more insights. Here are some snapshots from the paper:

Smear slides cropped dataset

Schematic diagram for extraction of cells ready to be sent to domain adaptation pipeline

Classification model used with different backbones

Results Table

Results Table

Results Table

Final output of the detection and segmentation pipeline for MaskRCNN and Domain Adaptation

Mask R-CNN losses

Domain Adaptation models

Results Table for Domain Adaptation

Domain Adaptation losses

Full pipeline

JSON outputs which can be used for automated annotation of new slides (Future work)

If you find this work useful, please consider citing

@article{PAL2024123660,
title = {Advancing instance segmentation and WBC classification in peripheral blood smear through domain adaptation: A study on PBC and the novel RV-PBS datasets},
journal = {Expert Systems with Applications},
volume = {249},
pages = {123660},
year = {2024},
issn = {0957-4174},
doi = {https://doi.org/10.1016/j.eswa.2024.123660},
url = {https://www.sciencedirect.com/science/article/pii/S0957417424005268},
author = {Jimut Bahan Pal and Aniket Bhattacharyea and Debasis Banerjee and Br. Tamal Maharaj},
keywords = {Automated blood test, Detection, Domain adaptation, Instance segmentation, Peripheral blood smear}
}

About

Code for the paper titled "Advancing instance segmentation and WBC classification in peripheral blood smear through domain adaptation: A study on PBC and the novel RV-PBS datasets" published on Elsevier's Expert Systems With Applications (ESWA) journal.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages