ISubGVQA.models.scene_graph_encoder

Classes

SceneGraphEncoder

Functions

get_gt_scene_graph_encoding_layer(num_node_features, ...)

Module Contents

class ISubGVQA.models.scene_graph_encoder.SceneGraphEncoder(hidden_dim, dist=False)

Bases: torch.nn.Module

scene_graphs_helper

sg_vocab

hidden_dim

dist = False

sg_emb_dim = 300

sg_vocab_embedding

scene_graph_encoding_layer

graph_layer_norm

bbox_encoding

feat_reduc

forward(x, edge_index, edge_attr, batch, explainer=False, explainer_stage=False, gt_scene_graphs=None)

ISubGVQA.models.scene_graph_encoder.get_gt_scene_graph_encoding_layer(num_node_features, num_edge_features, hidden_dim)