How to improve prediction?

[abdou31]

Hello, Thanks for this project, it help me to predict face ( if wear mask or not )
but I got some problems ( drawing two boxes of differents class on the same object ):

How can I improve the prediction?

[h33p]

It appears that the model needs more training data.

Otherwise, I would calculate the overlap between 2 boxes and if they are roughly the same, discard the one with lower confidence.

[abdou31]

Good idea, but do you have a piece of code of this idea???