π¬ YOLOv5 β Microparticle Detection & Classification
Automated detection and classification of microplastics from microscopy images. Developed for industrial deployment at IZONICS β achieving >90% accuracy in production.
π Model Summary
| Property | Details |
|---|---|
| ποΈ Architecture | YOLOv5 |
| π― Task | Object Detection + Classification |
| π Accuracy | >90% in production |
| β‘ Speedup | 30% faster than classical image processing pipeline |
| π Deployment | Fully industrialized β prototype to production |
| βοΈ Framework | PyTorch |
| π License | MIT |
π§ What This Model Does
This model detects and classifies microplastic particles in microscopy images, enabling automated environmental monitoring and industrial quality control.
It was developed at IZONICS as a complete refactoring of an existing classical computer vision pipeline into a deep learning architecture β resulting in higher reliability, better accuracy, and significantly faster inference in real production environments.
Origin: Extended from a Master's thesis (VIBOT program, UniversitΓ© de Bourgogne) into a fully deployed industrial product.
π Performance
| Metric | Score |
|---|---|
| Production accuracy | >90% |
| Processing speedup vs. classical pipeline | 30% |
| Deployment status | β Fully industrialized |
π How to Use
import torch
# Load the model
model = torch.hub.load("ultralytics/yolov5", "custom", path="best.pt")
model.conf = 0.35 # confidence threshold
# Run inference
results = model("microscopy_image.jpg")
# Show results
results.show()
# Get detections as dataframe
df = results.pandas().xyxy[0]
print(df)
ποΈ Training Data
- Domain: Environmental microscopy images of microplastic particles
- Task: Multi-class detection and classification of particle types
- Augmentation: Custom augmentation tailored for microscopy image characteristics
β οΈ Limitations
- Optimized for specific microscopy imaging conditions used at IZONICS
- Performance may vary on microscopy setups with different magnification or lighting
- Not validated for clinical or regulatory use
π Related Resources
- π€ Live Demo Space
- π» GitHub Repository
π€ Author
Martin Badrous β Computer Vision & Deep Learning Engineer
- Downloads last month
- 2