Quality control on production lines with Computer Vision and TinyML
This repository includes a jupyter notebook that presents a complete pipeline for:
EDA on image data
Data preparation and augmentation
Deep learning (CNN) models development for image classification with TensorFlow
Models evaluation
Model interpretation predictions with LIME
Transformation to TFLite format to allow their usage in embedded devices
Post-training quantization
Quantization aware training
Overall evaluation
Complete explanations of the above are available in the notebook, which is also available in PDF format.
Moreover, there is a python script that allows the real-time usage of the developed models on video streams.
Main Characteristic
- EDA on image data
- Data preparation and augmentation
- Deep learning (CNN) models development for image classification with TensorFlow
- Models evaluation
- Model interpretation predictions with LIME
- Transformation to TFLite format to allow their usage in embedded devices
- Post-training quantization
- Quantization aware training
- Overall evaluation
- Real-time inference
Research areas
Integrative AI
Business Categories
Manufacturing
Last updated
24.01.2023 - 10:26
Documents