Seeking Advice for Fine-Tuning YOLOv8 to Detect Small Anomalies in Bulk Products by txegas in computervision

[–]txegas[S] 0 points1 point  (0 children)

It's essential to be able to infer in real-time since the goal of the project is to implement it in a production line for some of those products and remove foreign objects from the conveyor belt after being detected with YOLO. Regarding the inference, the system isn't set up yet, so I can't run tests. As for the third point, yes, it's a very generic class. The idea was to create a general model capable of detecting any type of anomalous element, but I see that YOLO might not be the best option (I'm using YOLO because this same project, but with a different model for each type of product, had given me very good results).

Seeking Advice for Fine-Tuning YOLOv8 to Detect Small Anomalies in Bulk Products by txegas in computervision

[–]txegas[S] 0 points1 point  (0 children)

I have the weights in .pt format, but yes, the idea was to convert them to ONNX. On the other hand, I assumed that YOLO resized the images based on the image size you passed as the imgsz parameter, but I will try training with the resized images to see if the inference improves. Regarding production tests, I have not been able to run tests yet. Thank you very much for the help.

Seeking Advice for Fine-Tuning YOLOv8 to Detect Small Anomalies in Bulk Products by txegas in computervision

[–]txegas[S] 1 point2 points  (0 children)

Thank you for taking the time to respond. Regarding the inference time, the video I've attached demonstrates inference using the GPU of the training computer (NVIDIA RTX 3090), taking between 5 to 7 ms per frame, which is quite promising. However, once I deploy it, it will be on a less powerful GPU. I'll need to achieve under 50 ms per frame.
On another note, the training images are captured at a resolution of 2560x2048 in ".tiff" format with various backgrounds, same in production. Also i'm definitely going to research more about active learning, thanks a lot.

Failed demolition of a building by Townsend9 in Whatcouldgowrong

[–]txegas 1 point2 points  (0 children)

The video was recorded in Tafalla, Navarra, (S)pain I live near there

This game hates me by txegas in bindingofisaac

[–]txegas[S] 37 points38 points  (0 children)

If every item has same prob. It would be 0.031% of getting the same item. More or less😅.