Generation of microbial colonies dataset with deep learning style transfer

Abstract

We introduce an effective strategy to generate an annotated synthetic dataset of microbiological images of Petri dishes that can be used to train deep learning models in a fully supervised fashion. The developed generator employs traditional computer vision algorithms together with a neural style transfer method for data augmentation. We show that the method is able to synthesize a dataset of realistic looking images that can be used to train a neural network model capable of localising, segmenting, and classifying five different microbial species. Our method requires significantly fewer resources to obtain a useful dataset than collecting and labeling a whole large set of real images with annotations. We show that starting with only 100 real images, we can generate data to train a detector that achieves comparable results (detection mAP = 0.416, and counting MAE = 4.49) to the same detector but trained on a real, several dozen times bigger dataset (mAP = 0.520, MAE = 4.31), containing over 7k images. We prove the usefulness of the method in microbe detection and segmentation, but we expect that it is general and flexible and can also be applicable in other domains of science and industry to detect various objects.

arXiv comment13 pages, 9 figures, 2 tables

References in corpus (19)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
• scikit-image: Image processing in Python
• Generative Adversarial Network in Medical Imaging: A Review
• A Survey of Deep Learning-based Object Detection
• MMDetection: Open MMLab Detection Toolbox and Benchmark
• NiftyNet: a deep-learning platform for medical imaging
• Universal Style Transfer via Feature Transforms
• Early-detection and classification of live bacteria using time-lapse coherent imaging and deep learning
• Multi-style Generative Network for Real-time Transfer
• Skin Lesion Synthesis with Generative Adversarial Networks
• Melanoma Detection using Adversarial Training and Deep Transfer Learning
• Explainable Diabetic Retinopathy Detection and Retinal Image Generation
• Generative Adversarial Networks for Image-to-Image Translation on Multi-Contrast MR Images - A Comparison of CycleGAN and UNIT
• A Real-World Demonstration of Machine Learning Generalizability: Intracranial Hemorrhage Detection on Head CT
• Evaluating the Clinical Realism of Synthetic Chest X-Rays Generated Using Progressively Growing GANs
• Deep Negative Volume Segmentation
• High-Resolution Network for Photorealistic Style Transfer
• MammoGANesis: Controlled Generation of High-Resolution Mammograms for Radiology Education

Cited by in corpus (2)

• Self-Normalized Density Map (SNDM) for Counting Microbiological Objects
• Performing Video Frame Prediction of Microbial Growth with a Recurrent Neural Network