Python 2/3 compatible download_model_binary.py

willyd · willyd · commit 91c15e85124c · 2017-01-13T14:33:35.000-05:00
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.')
