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¶
- static get_backbone(batch_norm)[source]¶
Get pretrained VGG16 models for feature extraction.
- Returns
feature sequence
- training: bool¶