How to prevent box overlapping Yolo v9

Community Help
1

Hi everyone,

I’m currently working on a YOLOv9 model to detect stairs in architectural plans, and I’ve finally reached a point where the model is starting to deliver promising results, which feels fantastic!

However, in some cases, the model is generating overlapping bounding boxes, as shown below. I’ve tried adjusting the ‘iou’ parameter, but unfortunately, it hasn’t resolved the issue.

0011- PLAN R+1 COLLECTIF A
0011- PLAN R+1 COLLECTIF A1920×1356 274 KB

Does anyone have suggestions on how I can improve this?

Thanks in advance!

2

Hi @Igor_Barcelos - thanks for reaching out!

How many images is your model trained on? You may just need to add some more.

My other advice is using a piece of python code with the inference result that only takes the most confident prediction if two blocks are touching.

3

This topic was automatically closed 7 days after the last reply. New replies are no longer allowed.