Bering.models.ImageEncoder.get_conv2d_embedding

ImageEncoder.get_conv2d_embedding(x)[source]

Get the shared convolution embedding (as the input of SPP) of a large image for edges that derived from this image. input image shape: 1 * n_channels * h * w

Parameters:

x (Tensor) – input image tensor