Scene Graph prediction on custom video without downloading any dataset and using pre trained models

yrcong / STTran

Spatial-Temporal Transformer for Dynamic Scene Graph Generation, ICCV2021

MIT License

187 stars 34 forks source link

Scene Graph prediction on custom video without downloading any dataset and using pre trained models #33

Closed cspearl closed 2 years ago

cspearl commented 2 years ago

I actually just want to use pretrained models files and detect scene graphs in a custom video/images, instead of downloading AG dataset or training right now. But I am not able to make the necessary modifications in the code. If someone could please help me out.

yrcong commented 2 years ago

You could have a look at 29.