Scene-Graph-Benchmark.pytorch icon indicating copy to clipboard operation
Scene-Graph-Benchmark.pytorch copied to clipboard

Published 20 hours ago • verified publisher icon KaihuaTang

Transformer model details

Open coldmanck opened this issue 3 years ago • 2 comments

❓ Questions and Help

Hi @KaihuaTang, thanks for your awesome work! I am wondering where I can find the details or an illustration of the Transformer model, as I couldn't find it anywhere in this codebase and the (Unbiased SGG) paper. Do I miss any description about it?

coldmanck avatar Oct 25 '20 05:10 coldmanck

The code is here https://github.com/KaihuaTang/Scene-Graph-Benchmark.pytorch/blob/master/maskrcnn_benchmark/modeling/roi_heads/relation_head/model_transformer.py

In the paper, we don't use the transformer, since we need to improve "existing" methods.

KaihuaTang avatar Oct 25 '20 05:10 KaihuaTang

Thanks for your prompt reply! I understand the model is inside this code base, I was wondering is there any explanation about the architecture.

coldmanck avatar Oct 25 '20 07:10 coldmanck