can not find where to map original rel_label index to decending index?

dongxingning / SHA-GCL-for-SGG

Code for paper "Stacked Hybrid-Attention and Group Collaborative Learning for Unbiased Scene Graph Generation"

MIT License

32 stars 5 forks source link

can not find where to map original rel_label index to decending index? #3

Closed rafa-cxg closed 2 years ago

rafa-cxg commented 2 years ago

Hi, appreciate your work!

I am confused that the code seems to use the original rel_label class indexes as each group's label. I was thought that the relation labels should be rearranged by new order.

However, self.pre_group_matrix seems to directly assign 0-5 to 51 predicates, without descending order before assignment.

dongxingning commented 2 years ago

Thanks for your question! We indeed rearrange the rel_label class indexes in maskrcnn_benchmark/data/datasets/visual_genome.py [Line 389-392]. We hope you find it helpful. 1649646011(1)

rafa-cxg commented 2 years ago

yeah, thanks for that!