search

yikang-li / FactorizableNet

Factorizable Net (Multi-GPU version): An Efficient Subgraph-based Framework for Scene Graph Generation
216 stars 38 forks source link

How to compute the weight of predicate #10

Closed waallf closed 5 years ago

waallf commented 5 years ago

Hi,I want to know how to compute weight of predicate,which is in your inverse_weight of MSDN project.

yikang-li commented 5 years ago
  1. calculate the frequency of each class;
  2. the weight equals to the sqrt(1/freq)
  3. normalize the weight by scaling the smallest weight to 1

That's is just an empirical operation. We didn't perform any ablation study.