YOLOv7 drawing multiple bounding boxes over the same object

[ianhgenesys]

Hi,

I am facing some problems with the implementation of YOLOv7 on the Luxonis module. After following the instructions from https://github.com/luxonis/depthai-ml-training/blob/master/colab-notebooks/YoloV7_training.ipynb, I am able to get a working model on my computer using the testing module from https://github.com/WongKinYiu/yolov7. However, once I convert the file to a Blob and run it from Luxonis, I am getting three bounding boxes drawn over a single object, each centered around the object, an increasing in size.

Does anybody have a suggestion on what it is doing? I think it may be drawing all three bounding boxes obtained at each scale and plotting them overlapping each other? What should I do to circumvent this?

Thank you very much!

Kind regards, Ian

[Erol444]

Hi @ianhgenesys , Could you share an image of that? I would assume your IoU threshold is not confiugred properly and you should change it, but can't tell for sure without any additional info. Thanks, Erik