mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update imagenet weights path

pull/201/head
KaiyangZhou 2019-05-28 15:33:04 +01:00
parent 21171ebb1d
commit 9963cff629
3 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
torchreid/models/mlfn.py
View File

@ -12,7 +12,7 @@ import torch.utils.model_zoo as model_zoo
model_urls = {
# training epoch = 5, top1 = 51.6
'imagenet': 'http://www.eecs.qmul.ac.uk/~kz303/deep-person-reid/model-zoo/imagenet-pretrained/mlfn-9cb5a267.pth.tar',
'imagenet': 'https://mega.nz/#!YHxAhaxC!yu9E6zWl0x5zscSouTdbZu8gdFFytDdl-RAdD2DEfpk',
}
@ -209,5 +209,7 @@ def init_pretrained_weights(model, model_url):
def mlfn(num_classes, loss='softmax', pretrained=True, **kwargs):
model = MLFN(num_classes, loss, **kwargs)
if pretrained:
init_pretrained_weights(model, model_urls['imagenet'])
#init_pretrained_weights(model, model_urls['imagenet'])
import warnings
warnings.warn('The imagenet pretrained weights need to be manually downloaded from {}'.format(model_urls['imagenet']))
return model

12
torchreid/models/mobilenetv2.py
View File

@ -11,9 +11,9 @@ import torch.utils.model_zoo as model_zoo
model_urls = {
# 1.0: top-1 71.3
'mobilenetv2_1dot0': 'http://eecs.qmul.ac.uk/~kz303/deep-person-reid/model-zoo/imagenet-pretrained/mobilenetv2_1.0-0f96a698.pth',
'mobilenetv2_1dot0': 'https://mega.nz/#!NKp2wAIA!1NH1pbNzY_M2hVk_hdsxNM1NUOWvvGPHhaNr-fASF6c',
# 1.4: top-1 73.9
'mobilenetv2_1dot4': 'http://eecs.qmul.ac.uk/~kz303/deep-person-reid/model-zoo/imagenet-pretrained/mobilenetv2_1.4-bc1cc36b.pth',
'mobilenetv2_1dot4': 'https://mega.nz/#!RGhgEIwS!xN2s2ZdyqI6vQ3EwgmRXLEW3khr9tpXg96G9SUJugGk',
}
@ -210,7 +210,9 @@ def mobilenetv2_1dot0(num_classes, loss, pretrained=True, **kwargs):
**kwargs
)
if pretrained:
init_pretrained_weights(model, model_urls['mobilenetv2_1dot0'])
#init_pretrained_weights(model, model_urls['mobilenetv2_1dot0'])
import warnings
warnings.warn('The imagenet pretrained weights need to be manually downloaded from {}'.format(model_urls['mobilenetv2_1dot0']))
return model
@ -224,5 +226,7 @@ def mobilenetv2_1dot4(num_classes, loss, pretrained=True, **kwargs):
**kwargs
)
if pretrained:
init_pretrained_weights(model, model_urls['mobilenetv2_1dot4'])
#init_pretrained_weights(model, model_urls['mobilenetv2_1dot4'])
import warnings
warnings.warn('The imagenet pretrained weights need to be manually downloaded from {}'.format(model_urls['mobilenetv2_1dot4']))
return model

6
torchreid/models/shufflenet.py
View File

@ -12,7 +12,7 @@ import torch.utils.model_zoo as model_zoo
model_urls = {
# training epoch = 90, top1 = 61.8
'imagenet': 'http://www.eecs.qmul.ac.uk/~kz303/deep-person-reid/model-zoo/imagenet-pretrained/shufflenet-bee1b265.pth.tar',
'imagenet': 'https://mega.nz/#!RDpUlQCY!tr_5xBEkelzDjveIYBBcGcovNCOrgfiJO9kiidz9fZM',
}
@ -161,5 +161,7 @@ def init_pretrained_weights(model, model_url):
def shufflenet(num_classes, loss='softmax', pretrained=True, **kwargs):
model = ShuffleNet(num_classes, loss, **kwargs)
if pretrained:
init_pretrained_weights(model, model_urls['imagenet'])
#init_pretrained_weights(model, model_urls['imagenet'])
import warnings
warnings.warn('The imagenet pretrained weights need to be manually downloaded from {}'.format(model_urls['imagenet']))
return model