Enhancing Colony Detection of Microorganisms in Agar Dishes Using SAM-Based Synthetic Data Augmentation in Low-Data Scenarios
Source abstract
Abstract
In many medical and pharmaceutical processes, continuous hygiene monitoring relies on manual detection of microorganisms in agar dishes by skilled personnel. While deep learning offers the potential for automating this task, it often faces limitations due to insufficient training data, a common issue in colony detection. To address this, we propose a simple yet efficient SAM-based pipeline for Copy-Paste data augmentation to enhance detection performance, even with limited data. This paper explores a method where annotated microbial colonies from real images were copied and pasted into empty agar dish images to create new synthetic samples. These new samples inherited the annotations of the colonies inserted into them so that no further labeling was required. The resulting synthetic datasets were used to train a YOLOv8 detection model, which was then fine-tuned on just 10 to 1000 real images. The best fine-tuned model, trained on only 1000 real images, achieved an mAP of 60.6, while a base model trained on 5241 real images achieved 64.9. Although far fewer real images were used, the fine-tuned model performed comparably well, demonstrating the effectiveness of the SAM-based Copy-Paste augmentation. This approach matches or even exceeds the performance of the current state of the art in synthetic data generation in colony detection and can be expanded to include more microbial species and agar dishes.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Cell Image Analysis Techniques · Microbial Community Ecology and Physiology
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Research and Transfer Center CeMOS, Mannheim University of Applied Sciences, 68163 Mannheim, Germany
Funding
Federal Ministry of Education and Research Germany · German Research Foundation · Albert and Anneliese Konanz Foundation
References (49)
Show source-provided references
1. Geissmann, Q. (2013). OpenCFU, a new free and open-source software to count cell colonies and other circular objects. PLoS ONE, 8.10.1371/journal.pone.0054072 ↗
2. Khan, A.U.M., Torelli, A., Wolf, I., and Gretz, N. (2018). AutoCellSeg: Robust automatic colony forming unit (CFU)/cell analysis using adaptive image segmentation and easy-to-use post-editing techniques. Sci. Rep., 8.10.1038/s41598-018-24916-9 ↗
3. Pawłowski, J., Majchrowska, S., and Golan, T. (2022). Generation of microbial colonies dataset with deep learning style transfer. Sci. Rep., 12.10.1038/s41598-022-09264-z ↗
4. Majchrowska, S., Pawłowski, J., Guła, G., Bonus, T., Hanas, A., Loch, A., Pawlak, A., Roszkowiak, J., Golan, T., and Drulis-Kawa, Z. (2021). AGAR a microbial colony dataset for deep learning detection. arXiv.10.21203/rs.3.rs-668667/v1 ↗
5. Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A.N., Kaiser, Ł., and Polosukhin, I. (2017). Attention is all you need. Adv. Neural Inf. Process. Syst. (NeurIPS), 30.
6. Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., and Gelly, S. (May, January 26). An Image is Worth 16 × 16 Words: Transformers for Image Recognition at Scale. Proceedings of the International Conference on Learning Representations (ICLR), Virtual.
7. Ebert, N., Stricker, D., and Wasenmüller, O. (2023). PLG-ViT: Vision Transformer with Parallel Local and Global Self-Attention. Sensors, 23.10.3390/s23073447 ↗
8. Ebert, N., Stricker, D., and Wasenmüller, O. (2023, January 2–6). Transformer-based Detection of Microorganisms on High-Resolution Petri Dish Images. Proceedings of the International Conference on Computer Vision Workshops (ICCVW), Paris, France.10.1109/ICCVW60793.2023.00428 ↗
9. U-Net: Deep learning for cell counting, detection, and morphometry10.1038/s41592-018-0261-2 ↗
10. Naets, T., Huijsmans, M., Smyth, P., Sorber, L., and de Lannoy, G. (2021). A Mask R-CNN approach to counting bacterial colony forming units in pharmaceutical development. arXiv.
11. Medical image data augmentation: Techniques, comparisons and interpretations10.1007/s10462-023-10453-z ↗
12. Andreini, P., Bonechi, S., Bianchini, M., Mecocci, A., and Scarselli, F. (2020). Image generation by GAN and style transfer for agar plate image segmentation. Comput. Methods Programs Biomed., 184.10.1016/j.cmpb.2019.105268 ↗
13. Deepak, S., and Ameer, P. (2020, January 2–4). MSG-GAN based synthesis of brain MRI with meningioma for data augmentation. Proceedings of the International Conference on Electronics, Computing and Communication Technologies (CONECCT), Bangalore, India.10.1109/CONECCT50063.2020.9198672 ↗
14. Ghiasi, G., Cui, Y., Srinivas, A., Qian, R., Lin, T.Y., Cubuk, E.D., Le, Q.V., and Zoph, B. (2021, January 20–25). Simple copy-paste is a strong data augmentation method for instance segmentation. Proceedings of the Conference on Computer Vision and Pattern Recognition (CVPR), Nashville, TN, USA.10.1109/CVPR46437.2021.00294 ↗
15. Kirillov, A., Mintun, E., Ravi, N., Mao, H., Rolland, C., Gustafson, L., Xiao, T., Whitehead, S., Berg, A.C., and Lo, W.Y. (2023, January 2–6). Segment anything. Proceedings of the International Conference on Computer Vision (ICCV), Paris, France.10.1109/ICCV51070.2023.00371 ↗
16. Automatic counting of bacterial cultures—A new machine10.1109/IRET-ME.1958.5008510 ↗
17. Mansberg, H. (1957). Automatic particle and bacterial colony counter. Science, 126.10.1126/science.126.3278.823 ↗
18. Bacterial colony counting with convolutional neural networks in digital microbiology imaging10.1016/j.patcog.2016.07.016 ↗
19. Andreini, P., Bonechi, S., Bianchini, M., Mecocci, A., and Scarselli, F. (2018, January 4–7). A deep learning approach to bacterial colony segmentation. Proceedings of the International Conference on Artificial Neural Networks (ICANN), Rhodes, Greece.10.1007/978-3-030-01424-7_51 ↗
20. Cell segmentation using a similarity interface with a multi-task convolutional neural network10.1109/JBHI.2018.2885544 ↗
21. Ronneberger, O., Fischer, P., and Brox, T. (2015, January 5–9). U-net: Convolutional networks for biomedical image segmentation. Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI), Munich, Germany.10.1007/978-3-319-24574-4_28 ↗
22. He, K., Gkioxari, G., Dollár, P., and Girshick, R. (2017, January 22–29). Mask r-cnn. Proceedings of the International Conference on Computer Vision (ICCV), Venice, Italy.10.1109/ICCV.2017.322 ↗
23. Liu, S.J., Huang, P.C., Liu, X.S., Lin, J.J., and Zou, Z. (2022). A two-stage deep counting for bacterial colonies from multi-sources. Appl. Soft Comput., 130.10.1016/j.asoc.2022.109706 ↗
24. Cai, Z., and Vasconcelos, N. (2018, January 18–22). Cascade r-cnn: Delving into high quality object detection. Proceedings of the Conference on Computer Vision and Pattern Recognition (CVPR), Salt Lake City, UT, USA.10.1109/CVPR.2018.00644 ↗
25. Ren, S., He, K., Girshick, R., and Sun, J. (2017). Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks. Trans. Pattern Anal. Mach. Intell., 39.6.10.1109/TPAMI.2016.2577031 ↗
26. Liu, S., Qi, L., Qin, H., Shi, J., and Jia, J. (2018, January 18–22). Path aggregation network for instance segmentation. Proceedings of the Conference on Computer Vision and Pattern Recognition (CVPR), Salt Lake City, UT, USA.10.1109/CVPR.2018.00913 ↗
27. Feng, C., Zhong, Y., Gao, Y., Scott, M.R., and Huang, W. (2021, January 10–17). Tood: Task-aligned one-stage object detection. Proceedings of the International Conference on Computer Vision (ICCV), Montreal, QC, Canada.10.1109/ICCV48922.2021.00349 ↗
28. Shorten, C., and Khoshgoftaar, T.M. (2019). A survey on image data augmentation for deep learning. J. Big Data, 6.10.1186/s40537-019-0197-0 ↗
29. Cubuk, E.D., Zoph, B., Shlens, J., and Le, Q.V. (2020, January 13–19). Randaugment: Practical automated data augmentation with a reduced search space. Proceedings of the Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), Seattle, WA, USA.10.1109/CVPRW50498.2020.00359 ↗
30. Hendrycks, D., Mu, N., Cubuk, E.D., Zoph, B., Gilmer, J., and Lakshminarayanan, B. (May, January 26). AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty. Proceedings of the International Conference on Learning Representations (ICLR), Virtual.
31. Zhang, H., Cisse, M., Dauphin, Y.N., and Lopez-Paz, D. (May, January 30). mixup: Beyond Empirical Risk Minimization. Proceedings of the International Conference on Learning Representations, Vancouver, BC, Canada.
32. Yun, S., Han, D., Oh, S.J., Chun, S., Choe, J., and Yoo, Y. (November, January 27). Cutmix: Regularization strategy to train strong classifiers with localizable features. Proceedings of the International Conference on Computer Vision (ICCV), Seoul, Republic of Korea.
33. Bochkovskiy, A., Wang, C.Y., and Liao, H.Y.M. (2020). Yolov4: Optimal speed and accuracy of object detection. arXiv.
34. Oehri, S., Ebert, N., Abdullah, A., Stricker, D., and Wasenmüller, O. (2024, January 1–5). GenFormer—Generated Images are All You Need to Improve Robustness of Transformers on Small Datasets. Proceedings of the International Conference on Pattern Recognition (ICPR), Kolkata, India.10.1007/978-3-031-78166-7_12 ↗
35. Sarıyıldız, M.B., Alahari, K., Larlus, D., and Kalantidis, Y. (2023, January 17–24). Fake it till you make it: Learning transferable representations from synthetic ImageNet clones. Proceedings of the Conference on Computer Vision and Pattern Recognition (CVPR), Vancouver, BC, Canada.10.1109/CVPR52729.2023.00774 ↗
36. Reichardt, L., Uhr, L., and Wasenmüller, O. (2024, January 14–18). Text3DAug – Prompted Instance Augmentation for LiDAR Perception. Proceedings of the International Conference on Intelligent Robots and Systems (IROS), Abu Dhabi, United Arab Emirates.10.1109/IROS58592.2024.10802726 ↗
37. Generative adversarial networks10.1145/3422622 ↗
38. Alyafi, B., Diaz, O., and Marti, R. (2020, January 15–20). DCGANs for realistic breast mass augmentation in x-ray mammography. Proceedings of the Medical Imaging 2020: Computer-Aided Diagnosis, Houston, TX, USA.10.1117/12.2543506 ↗
39. Desai, S.D., Giraddi, S., Verma, N., Gupta, P., and Ramya, S. (2020, January 25–26). Breast cancer detection using gan for limited labeled dataset. Proceedings of the International Conference on Computational Intelligence and Communication Networks (CICN), Bhimtal, India.10.1109/CICN49253.2020.9242551 ↗
40. Shen, T., Hao, K., Gou, C., and Wang, F.Y. (2021). Mass image synthesis in mammogram with contextual information based on GANs. Comput. Methods Programs Biomed., 202.10.1016/j.cmpb.2021.106019 ↗
41. Li, Q., Yu, Z., Wang, Y., and Zheng, H. (2020). TumorGAN: A multi-modal data augmentation framework for brain tumor segmentation. Sensors, 20.10.3390/s20154203 ↗
42. Neelima, G., Chigurukota, D.R., Maram, B., and Girirajan, B. (2022). Optimal DeepMRSeg based tumor segmentation with GAN for brain tumor classification. Biomed. Signal Process. Control, 74.10.1016/j.bspc.2022.103537 ↗
43. Bissoto, A., Perez, F., Valle, E., and Avila, S. (2018, January 16–20). Skin lesion synthesis with generative adversarial networks. Proceedings of the OR 2.0 Context-Aware Operating Theaters, Computer Assisted Robotic Endoscopy, Clinical Image-Based Procedures, and Skin Image Analysis, Granada, Spain.10.1007/978-3-030-01201-4_32 ↗
44. Ghorbani, A., Natarajan, V., Coz, D., and Liu, Y. (2020, January 11). Dermgan: Synthetic generation of clinical skin images with pathology. Proceedings of the Machine Learning for Health Workshop, Virtual.
45. Dar, S.U.H., Ghanaat, A., Kahmann, J., Ayx, I., Papavassiliou, T., Schoenberg, S.O., and Engelhardt, S. (2023). Investigating data memorization in 3d latent diffusion models for medical image synthesis. arXiv.10.1007/978-3-031-53767-7_6 ↗
46. Sagers, L.W., Diao, J.A., Melas-Kyriazi, L., Groh, M., Rajpurkar, P., Adamson, A.S., Rotemberg, V., Daneshjou, R., and Manrai, A.K. (2023). Augmenting medical image classifiers with synthetic data from latent diffusion models. arXiv.
47. Gulrajani, I., Ahmed, F., Arjovsky, M., Dumoulin, V., and Courville, A.C. (2017). Improved training of wasserstein gans. Adv. Neural Inf. Process. Syst. (NeurIPS), 360.
48. Jocher, G., Chaurasia, A., and Qiu, J. (2024, December 16). Ultralytics YOLOv8. Available online: https://github.com/ultralytics/ultralytics.
49. Lin, T.Y., Maire, M., Belongie, S., Hays, J., Perona, P., Ramanan, D., Dollár, P., and Zitnick, C.L. (2014, January 6–12). Microsoft coco: Common objects in context. Proceedings of the European Conference on Computer Vision (ECCV), Zurich, Switzerland.10.1007/978-3-319-10602-1_48 ↗