Bering.models.ImageEncoder.forward

ImageEncoder.forward(images)

Run SPP layer and FC layers for edges with convolutional embedding features

Parameters:

images (Tensor) – image embeddings (n_edges x n_conv2d_embeddings x W x H) for all edges as the input of SPP and FC layers