MILVLG / bottom-up-attention.pytorch

A PyTorch reimplementation of bottom-up-attention models
Apache License 2.0
294 stars 76 forks source link

About relationship detection #87

Closed ezeli closed 2 years ago

ezeli commented 2 years ago

Hi, Thanks for your excellent work. I want to extract the relationship and don't know how to do it?

MIL-VLG commented 2 years ago

What do you mean by the relationship? Current models only support object and attribute prediction. We (Neither did the original caffe repo) do not support the relationship prediction.

ezeli commented 2 years ago

OK, thanks!