How to improve prediction?

experiencor / keras-yolo3

Training and Detecting Objects with YOLO3

MIT License

1.6k stars 862 forks source link

How to improve prediction? #278

Open abdou31 opened 4 years ago

abdou31 commented 4 years ago

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 commented 4 years ago

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 commented 4 years ago

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