Upload models with huggingface_hub

models/download_models.py

@@ -0,0 +1,31 @@
+import torch
+from os import path
+from sys import version_info
+from collections import OrderedDict
+from torch.utils.model_zoo import load_url
+
+
+# Download the VGG-19 model and fix the layer names
+print("Downloading the VGG-19 model")
+sd = load_url("https://web.eecs.umich.edu/~justincj/models/vgg19-d01eb7cb.pth")
+map = {'classifier.1.weight':u'classifier.0.weight', 'classifier.1.bias':u'classifier.0.bias', 'classifier.4.weight':u'classifier.3.weight', 'classifier.4.bias':u'classifier.3.bias'}
+sd = OrderedDict([(map[k] if k in map else k,v) for k,v in sd.items()])
+torch.save(sd, path.join("models", "vgg19-d01eb7cb.pth"))
+
+# Download the VGG-16 model and fix the layer names
+print("Downloading the VGG-16 model")
+sd = load_url("https://web.eecs.umich.edu/~justincj/models/vgg16-00b39a1b.pth")
+map = {'classifier.1.weight':u'classifier.0.weight', 'classifier.1.bias':u'classifier.0.bias', 'classifier.4.weight':u'classifier.3.weight', 'classifier.4.bias':u'classifier.3.bias'}
+sd = OrderedDict([(map[k] if k in map else k,v) for k,v in sd.items()])
+torch.save(sd, path.join("models", "vgg16-00b39a1b.pth"))
+
+# Download the NIN model
+print("Downloading the NIN model")
+if version_info[0] < 3:
+    import urllib
+    urllib.URLopener().retrieve("https://raw.githubusercontent.com/ProGamerGov/pytorch-nin/master/nin_imagenet.pth", path.join("models", "nin_imagenet.pth"))
+else: 
+    import urllib.request
+    urllib.request.urlretrieve("https://raw.githubusercontent.com/ProGamerGov/pytorch-nin/master/nin_imagenet.pth", path.join("models", "nin_imagenet.pth"))
+
+print("All models have been successfully downloaded")

models/nin_imagenet.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:40e4a6ae31ab31cb2f50fb45c9974fc7b5530d716899c77196cd01b61f33160e
+size 30383343

models/vgg16-00b39a1b.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2007cdad2519a9bbecdf5cd3c54145237502349a194e0d0a41a826e5a277c187
+size 553438853

models/vgg19-d01eb7cb.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dd893d447ece014ad32d71b8d32b1e0b5c09b912a2866990c286fdeebd368d74
+size 574679411