vgg16 test

[kruzel]

Hi,
do you support vgg16 pretrained test?
if yes how can I run it?
what *.pth should I use?

parser.add_argument('--network', '-n', metavar='NETWORK', default='resnet101-solar-best.pth',
help="network to be evaluated. " )

alternatively,
can you explain how to train with vgg16?
I tried setting --arch 'vgg16'
but it fails in networks.py in init() in this line:
last_feat_in = base_model.inplanes
orch.nn.modules.module.ModuleAttributeError: 'VGG' object has no attribute 'inplanes'

Thanks

[tonyngjichun]

Hi, thanks for the question and apologies for the delayed response. In this work we focused on ResNet models at it was shown that ResNet consistently outperforms VGG in image retrieval benchmarks, hence the models were designed around ResNet. The base_model.inplanes is the feature size from the last feature maps, for ResNet101 this is 2048. For all VGG models, this number would be 512. Hope this helps!