VGG16_base

class src.backbone.VGG16_base(batch_norm=True, final_layers=False)[source]

The base class of VGG16. It downloads the pretrained weight by torchvision API, and maintain the layers needed for deep graph matching models.

property device
forward(*input)[source]
static get_backbone(batch_norm)[source]

Get pretrained VGG16 models for feature extraction.

Returns

feature sequence

training: bool