From 9963cff6292093ceb3c25a2a48579266deb3d199 Mon Sep 17 00:00:00 2001
From: KaiyangZhou <k.zhou@qmul.ac.uk>
Date: Tue, 28 May 2019 15:33:04 +0100
Subject: [PATCH] update imagenet weights path

---
 torchreid/models/mlfn.py        |  6 ++++--
 torchreid/models/mobilenetv2.py | 12 ++++++++----
 torchreid/models/shufflenet.py  |  6 ++++--
 3 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/torchreid/models/mlfn.py b/torchreid/models/mlfn.py
index 5a4e2b6..0c9353f 100644
--- a/torchreid/models/mlfn.py
+++ b/torchreid/models/mlfn.py
@@ -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
\ No newline at end of file
diff --git a/torchreid/models/mobilenetv2.py b/torchreid/models/mobilenetv2.py
index 8d48335..6ca8bdb 100644
--- a/torchreid/models/mobilenetv2.py
+++ b/torchreid/models/mobilenetv2.py
@@ -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
\ No newline at end of file
diff --git a/torchreid/models/shufflenet.py b/torchreid/models/shufflenet.py
index 8367257..911a3e0 100644
--- a/torchreid/models/shufflenet.py
+++ b/torchreid/models/shufflenet.py
@@ -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
\ No newline at end of file