End-to-end MLOps pipeline to detect 61 types of electronic components from images — powered by YOLOv5s, served via Flask, containerized with Docker, and automated through GitHub Actions CI/CD.
ElectroCom-61 detects electronics parts like resistors, capacitors, sensors, Arduino boards, LEDs, and more — directly from images through a Flask web interface.
Use cases:
- ✅ PCB assembly verification
- ✅ Electronics inventory management
- ✅ Automated quality inspection
- ✅ Hobby & educational projects
Components_Identification_system/
├── componentsDetection/ # Core pipeline package (ingestion, training, evaluation)
├── data/ # Dataset (train / valid / test)
├── artifacts/ # Model weights & training outputs
├── yolov5/ # YOLOv5 submodule
├── templates/ # Flask HTML templates
├── research/ # Experiments & notebooks
├── validation/ # Validation scripts & results
├── log/ # Runtime logs
├── app.py # Flask web app entry point
├── Dockerfile # Container definition
├── requirements.txt # Dependencies
└── setup.py # Package installer
Data Ingestion → Preprocessing → Training → Evaluation → Flask App → Docker → CI/CD
| Stage | Description |
|---|---|
| Data Ingestion | Downloads and organizes the 61-class dataset |
| Preprocessing | Validates annotations and applies augmentation |
| Training | Fine-tunes YOLOv5s on the component dataset |
| Evaluation | Reports mAP, Precision, Recall, and Confusion Matrix |
| Flask App | Serves predictions via a web interface |
| Docker | Packages the full app into a portable container |
| CI/CD | GitHub Actions auto-builds and tests on every push |
# Clone the repo
git clone https://github.com/your-username/Components_Identification_system.git
cd Components_Identification_system
# Install dependencies
pip install -r requirements.txt
python setup.py install
# Run the app
python app.py
# OR with Docker
docker build -t componentsDetection .
docker run -p 8080:8080 componentsDetection| Property | Value |
|---|---|
| Architecture | YOLOv5s |
| Classes | 61 |
| Input Size | 640 × 640 |
| Serving | Flask |
| Container | Docker |
| CI/CD | GitHub Actions |