search

wgcban / Metric-CD

Official PyTorch implementation of Deep Metric Learning for Unsupervised Change Detection in Remote Sensing Images
https://arxiv.org/abs/2303.09536
15 stars 3 forks source link

Can provide a download address for pre training weights #2

Open WesternTrail opened 1 year ago

WesternTrail commented 1 year ago

Thank you for your work,！and may I ask if the following uses are included in your code self.blocks.load_state_dict(torch.load("/media/lidan/ssd/Chaminda/change_detection/VGG_finetune_AID/saved_model/vgg16_AID_049fc.pth"), strict=False) Can you provide a download address for reading the pre training weight code.

abdulbb commented 1 year ago

Is there any update on this?

wgcban commented 1 year ago

model_urls = { 'vgg11': 'https://download.pytorch.org/models/vgg11-bbd30ac9.pth', 'vgg13': 'https://download.pytorch.org/models/vgg13-c768596a.pth', 'vgg16': 'https://download.pytorch.org/models/vgg16-397923af.pth', 'vgg19': 'https://download.pytorch.org/models/vgg19-dcbb9e9d.pth', 'vgg11_bn': 'https://download.pytorch.org/models/vgg11_bn-6002323d.pth', 'vgg13_bn': 'https://download.pytorch.org/models/vgg13_bn-abd245e5.pth', 'vgg16_bn': 'https://download.pytorch.org/models/vgg16_bn-6c64b313.pth', 'vgg19_bn': 'https://download.pytorch.org/models/vgg19_bn-c79401a0.pth', }

wgcban commented 1 year ago

@abdulbb @WesternTrail

wgcban commented 1 year ago

https://github.com/minar09/VGG16-PyTorch/blob/master/vgg.py