orrzohar / PROB

[CVPR 2023] Official Pytorch code for PROB: Probabilistic Objectness for Open World Object Detection
Apache License 2.0
110 stars 16 forks source link

Model selection #40

Closed Rzx520 closed 1 year ago

Rzx520 commented 1 year ago

I am happy to help! Yes, these are very similar to the curves I got on my machine: image

I actually looked into why this happens and this has to do with PROB learning a good representation of objectness very early on (which is why U-Recall initially jumps, if you plot U-Recall inside epoch 1 you will see it increase from ~0 to 19). Then, as training progresses, it starts declining as it starts making more known object predictions, and therefore less unknown object predictions (e.g., ~U-Recall@100 goes down to ~U-Recall@80).

I will update the readme with this hyper parameter setup & machine type for future users.

If you encounter any new issues - do not hesitate to reach out, Orr

It is indeed like this. We can see that the U_R50 has decreased even after the training time has increased. I am quite puzzled, so why not choose the model with the highest U_R50? @orrzohar

orrzohar commented 1 year ago

Hi @Rzx520, Notice there is a balance here between K_AP50 and U_R50. I chose the checkpoint I thought was best --although this is somewhat subjective.

You can select the model with the highest U_R, the them K_AP wouldn't be as high

Best, Orr

Rzx520 commented 1 year ago

I understand. Thanks