Skip to content

ArdeleanRichard/YOLO-Histology

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

50 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Object detection in histology

A multi-dataset benchmark and test-time inference

This repository contains the code and configurations for a study submitted as a scientific research article.

📄 Overview

This project benchmarks modern object detection models (YOLOv8–v12, YOLOE, YOLO-World, RT-DETR) for Histopathology analysis. It evaluates their ability to detect retinal biomarkers across two publicly available datasets:

  • BCNB,
  • Nuclei,
  • TNBC,
  • MoNuSAC,
  • CryoNuSeg,

Our experiments identify YOLOE and YOLOv11 as the most effective models, offering an excellent trade-off between detection accuracy and computational efficiency.

Moreover, we propose Test-time Graph Similarity Propagation, a new method for test-time inference that improves upon the traditional confidence threshold and the TBSP algorithm.


📊 Datasets

This project requires a .yaml file to be created for the models to run on each dataset, as shown in this example:

train:  ..\..\data\MoNuSAC\images\train
val:    ..\..\data\MoNuSAC\images\val
test:   ..\..\data\MoNuSAC\images\test

nc: 4  # number of classes
names: ['Epithelial', 'Lymphocyte', 'Neutrophil', 'Macrophage']  # class names

1. BCNB Dataset

  • Name: BCNB
  • Description: digitized H&E-stained WSIs of primary tumor core-needle biopsy (CNB) specimens for predicting axillary lymph node (ALN) metastasis in early breast cancer (EBC). This extensive collection includes data from 1,058 EBC patients with clinically negative ALN, enrolled between May 2010 and August 2020.
  • Access: Openly available. 🔗 BCNB Dataset Page

2. Nuclei Dataset

  • Name: Nuclei
  • Description: 143 2,000 x 2,000 images ER+ BCa images scanned at 40x with 12,000 nuclei manually segmented across 137 patients.
  • Access: Openly available. 🔗 Nuclei Dataset Page

3. TNBC Dataset

  • Name: TNBC
  • Description: 50 images from 11 patients with a total of 4,022 annotated cells from H&E stained histological slides at 40x magnification. These slides were sourced from Triple Negative Breast Cancer (TNBC) patients
  • Access: Openly available. 🔗 TNBC Dataset Page

4. MoNuSAC Dataset

  • Name: MoNuSAC
  • Description: multi-organ dataset specifically designed for nuclei segmentation in histopathological images to encourage the computer vision research community to develop and test algorithms for detecting, segmenting, and classifying nuclei, tasks crucial for characterizing the tumor microenvironment (TME) in cancer prognostication and research. This large and diverse dataset comprises over 46,000 hand-annotated nuclei, sourced from 71 patients across 31 hospitals. It includes nuclei from four different organs and four distinct nucleus types: epithelial cells, lymphocytes, macrophages, and neutrophils.
  • Access: Openly available. 🔗 MoNuSAC Dataset Page

5. CryoNuSeg Dataset

  • Name: CryoNuSeg
  • Description: manually annotated nuclei instance segmentation dataset derived exclusively from frozen section (FS) H&E-stained images. CryoNuSeg comprises images from 10 human organs that were not exploited in other publicly available datasets, specifically the adrenal gland, larynx, lymph node, mediastinum, pancreas, pleura, skin, testis, thymus, and thyroid gland. It includes 30 whole slide images (WSIs), from which 512x512 pixel image patches acquired at 40x magnification were extracted, with one patch per WSI.
  • Access: Openly available. 🔗 CryoNuSeg Dataset Page

🧠 Models Evaluated

  • YOLOv8 to YOLOv12
  • YOLOE: Prompt-guided detection
  • YOLO-World: Open-vocabulary object detection
  • RT-DETR: Transformer-based object detection

📈 Results Summary

Inferencer Model Dataset F1 Score
Conf>0.5 RT-DETR TNBC 0.000
TSBP RT-DETR TNBC 0.000
TGSP RT-DETR TNBC 0.355
Conf>0.5 YOLOE nuclei 0.182
TSBP YOLOE nuclei 0.330
TGSP YOLOE nuclei 0.456
Conf>0.5 YOLOv11 nuclei 0.791
TSBP YOLOv11 nuclei 0.790
TGSP YOLOv11 nuclei 0.878

📬 Contact

For questions, please contact: 📧 ardeleaneugenrichard@gmail.com


About

This project benchmarks modern object detection models (YOLOv8–v12, YOLOE, YOLO-World, RT-DETR) for Histopathology analysis. Also the implementation of Test-time Graph Similarity Propagation algorithm (TSGP) can be found here.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages