search

OreoChocolate / MUREN

The official code for Relational Context Learning for Human-Object Interaction Detection, CVPR2023.
http://cvlab.postech.ac.kr/research/MUREN/
48 stars 5 forks source link

Focal loss for interaction classification #7

Closed xyc0833 closed 7 months ago

xyc0833 commented 8 months ago

image Thank you very much for your work. According to your model diagram, Human loss corresponds to the yellow box and object loss corresponds to the red box. Then how is this interaction loss designed? I see that you use Focal loss for interaction classification in your paper. I would like to know where the code of this Focal loss part is implemented in github. thank you.

OreoChocolate commented 7 months ago

Thank you for your interest in our research.

You can find our implementation at https://github.com/OreoChocolate/MUREN/blob/d7c8e63fad33eb0bd846dd36b3a8dbe536d8a39b/models/hoi.py#L168