CytoSPM

Model weights of CytoSPM: Open-Vocabulary Cytopathology Detection with Structured Prompt Bank (BIBM 2026).

Files

File Backbone Base AP / AP50 / AP75 Novel AP / AP50 / AP75 SHA256
cytospm_convnext_t.pth ConvNeXt-T 33.54 / 52.20 / 38.16 15.00 / 21.00 / 17.10 bc94215314319f712c731e8f5f24388df29d254cc2029695d32e815efb3aac4d

This is the model reported in the paper (Table III). The numbers above were measured with the released code on an RTX 3090; the paper values (33.55 / 52.20 / 38.17 and 15.00 / 21.00 / 17.20) were measured on an RTX 4090. Base AP is the COCO AP over the 24 base categories of PentaCyto; novel AP is the zero-shot COCO AP over the 9 held-out categories.

The file contains model weights only (state_dict) and loads with torch.load(..., weights_only=True).

Usage

git clone https://github.com/CVIU-CSU/CytoSPM.git && cd CytoSPM
CKPT_FILES=cytospm_convnext_t.pth bash tools/fetch_checkpoints.sh   # downloads and checks SHA256
python tools/test.py configs/cytospm.py checkpoints/cytospm_convnext_t.pth
python tools/test_novel.py configs/cytospm.py checkpoints/cytospm_convnext_t.pth

See the code repository for installation, data preparation and training. The PentaCyto images and annotations are not included.

Citation

@inproceedings{li2026cytospm,
  title     = {CytoSPM: Open-Vocabulary Cytopathology Detection with Structured Prompt Bank},
  author    = {Li, Wenjie and Xu, Zishan and Huang, Jinyang and Nie, Zhengxin and Kan, Shichao and Liang, Yixiong},
  booktitle = {IEEE International Conference on Bioinformatics and Biomedicine (BIBM)},
  year      = {2026}
}
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Paper for Doilion111/CytoSPM