search

JosephKJ / OWOD

(CVPR 2021 Oral) Open World Object Detection
https://josephkj.in
Apache License 2.0
1.04k stars 155 forks source link

Where is the code of "4.3. Energy based unknown identifier" #78

Closed lsqxxx closed 2 years ago

lsqxxx commented 2 years ago

Hello, author. First of all, thank you very much. I'd like to ask you about "4.3. Energy based unknown identifier", which contains the code about energy. I can't find it.

JosephKJ commented 2 years ago

Hi @lsqxxx,

Saving the data to learn the energy distribution: https://github.com/JosephKJ/OWOD/blob/master/detectron2/modeling/roi_heads/roi_heads.py#L455 (initiated from *_val.yaml)

Learning the energy distribution: https://github.com/JosephKJ/OWOD/blob/master/detectron2/engine/train_loop.py#L172

Using the energy distribution: https://github.com/JosephKJ/OWOD/blob/master/detectron2/evaluation/pascal_voc_evaluation.py#L122

Thanks, Joseph