enrypiff / buns_counter

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Buns counter

In this project I've started from an isssue saw during my as automation engineer. It's difficult in automated industrial production line to use just sensors when the enviroment is very dynamic, so I've decided to use the power of YOLO in order to fix this problem.

I manually annotated 12 photos I took during the commissionig of the production line and augmented them with Roboflow to have some some samples for my dataset.

alt text

Then I trained YOLO v8 for instance segmentation (I noticed that is more faster and accurate for very small datasets and in the application I used just the bounding boxes.

To handle the tracking problem for the counter I've used the SORT tracking algorithm.

This is the final result, a very promising use of YOLO in industrial field.

video_bun_result.mp4

About


Languages

Language:Python 75.6%Language:Jupyter Notebook 24.4%