CoolFace
Modelpublic

mziarehman4353/cell-segmentation-bbbc038

sourceHugging Facemitupdated 5mo agoView on Hugging Face
0likes
Model Card

๐Ÿงฌ Cell Segmentation & Classification (BBBC038)

This model performs:

  • โ€”๐Ÿ”น Cell segmentation using DeepLabV3+
  • โ€”๐Ÿ”น Cell classification (small vs large cells)

๐Ÿ“Š Results

  • โ€”Dice Score: 0.85
  • โ€”IoU Score: 0.74
  • โ€”Classification Accuracy: 91%

๐Ÿง  Model

  • โ€”Backbone: ResNet34
  • โ€”Architecture: DeepLabV3+

๐Ÿš€ Usage

python
import torch
import segmentation_models_pytorch as smp

model = smp.DeepLabV3Plus(
    encoder_name="resnet34",
    encoder_weights=None,
    classes=1
)

model.load_state_dict(torch.load("deeplabv3plus.pth"))
model.eval()
๐Ÿ“ Dataset
BBBC038 (Broad Bioimage Benchmark Collection)
๐Ÿ‘จโ€๐Ÿ’ป Author

Zia Ul Rehman Zafar