Python 2/3 compatible download_model_binary.py

BVLC · Jan 13, 2017 · 91c15e8 · 91c15e8
1 parent 365ac88
commit 91c15e8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/scripts/download_model_binary.py b/scripts/download_model_binary.py
@@ -3,10 +3,11 @@
 import sys
 import time
 import yaml
-import urllib
 import hashlib
 import argparse
 
+from six.moves import urllib
+
 required_keys = ['caffemodel', 'caffemodel_url', 'sha1']
 
 
@@ -69,7 +70,7 @@ def model_checks_out(filename=model_filename, sha1=frontmatter['sha1']):
         sys.exit(0)
 
     # Download and verify model.
-    urllib.urlretrieve(
+    urllib.request.urlretrieve(
         frontmatter['caffemodel_url'], model_filename, reporthook)
     if not model_checks_out():
         print('ERROR: model did not download correctly! Run this again.')