Colony Grounded SAM2: Zero-shot detection and segmentation of bacterial colonies using foundation models
Organizations: Wageningen Food Safety Research, Wageningen, The Netherlands
Abstract
The detection and classification of bacterial colonies in images of agar-plates is important in microbiology, but is hindered by the lack of labeled datasets. Therefore, we propose Colony Grounded SAM2, a zero-shot inference pipeline to detect and segment bacterial colonies in multiple settings without any further training. By utilizing the pre-trained foundation models Grounding DINO and Segment Anything Model 2, fine-tuned to the microbiological domain, we developed a model that is robust to data changes. Results showed a mean Average Precision of 93.1% and a score of 0.85, showing excellent detection and segmentation capabilities on out-of-distribution datasets. The entire pipeline with model weights are shared open access to aid with annotation- and classification purposes in microbiology.
Figures & tables
| Dataset name | # photos | # colonies | Type of ground truth |
|---|---|---|---|
| AGAR [ 5 ] | 18 000 | 336 442 | Bounding boxes |
| ADBC [ 6 ] | 369 | 56 865 | Bounding boxes |
| Hemolysis segmentation [ 9 ] | 286 | Unknown | Segmentations |
| Model | mAP/% |
|---|---|
| Faster R-CNN | 76.5 |
| YOLOv5n | 82.6 |
| YOLOv8n | 86.7 |
| YOLOv10n | 87.4 |
| Colony-YOLO | 91.1 |
| Colony Grounded SAM2 | 93.1* |