rhgao / co-separation

Co-Separating Sounds of Visual Objects (ICCV 2019)
Creative Commons Attribution 4.0 International
92 stars 23 forks source link

Select object with higher confidence score #14

Closed Huy0110 closed 2 years ago

Huy0110 commented 2 years ago

After reading the test.py file, I realized that if in a frame there are 2 objects corresponding to 2 instruments, the model will accept the object with higher confidence score as input. I wonder if this affects the loss function. I also read the train.py file, but I don't see the code where you choose the object with a higher confidence score. So can you tell me where it is or is it really working as I think or am I misunderstanding.

Thank you so much !!!