A new codebase for popular Scene Graph Generation methods (2020). Visualization & Scene Graph Extraction on custom images/datasets are provided. It's also a PyTorch implementation of paper “Unbiased Scene Graph Generation from Biased Training CVPR 2020”
MIT License
1.03k
stars
228
forks
source link
How to turn off the attribute head in both detector pretraining part and relationship prediction? #212
I want to use my own dataset but the data has no attributes, how to turn off the attribute head in both detector pretraining part and relationship prediction?
❓ Questions and Help
I want to use my own dataset but the data has no attributes, how to turn off the attribute head in both detector pretraining part and relationship prediction?