Fix py2 py3 incompatibility problem when converting models.

Author: ruotianluo

tools/convert_from_tensorflow.py

@@ -18,11 +18,11 @@
 
 del var_dict['Variable']
 
-for k in var_dict.keys():
+for k in list(var_dict.keys()):
     if 'Momentum' in k:
         del var_dict[k]
 
-for k in var_dict.keys():
+for k in list(var_dict.keys()):
     if k.find('/') >= 0:
         var_dict['resnet' + k[k.find('/'):]] = var_dict[k]
         del var_dict[k]
@@ -49,13 +49,13 @@
                 ('/', '.')])
 
 for a, b in dummy_replace.items():
-    for k in var_dict.keys():
+    for k in list(var_dict.keys()):
         if a in k:
             var_dict[k.replace(a,b)] = var_dict[k]
             del var_dict[k]
 
 
-for k in var_dict.keys():
+for k in list(var_dict.keys()):
     if 'unit_' in k:
         m = re.search('unit_(\d+)', k)
         var_dict[k.replace(m.group(0), str(int(m.group(1)) - 1))] = var_dict[k]

tools/convert_from_tensorflow_vgg.py

@@ -18,11 +18,11 @@
 
 del var_dict['Variable']
 
-for k in var_dict.keys():
+for k in list(var_dict.keys()):
     if 'Momentum' in k:
         del var_dict[k]
 
-for k in var_dict.keys():
+for k in list(var_dict.keys()):
     if k.find('/') >= 0:
         var_dict['vgg' + k[k.find('/'):]] = var_dict[k]
         del var_dict[k]
@@ -38,7 +38,7 @@
                 ('/', '.')])
 
 for a, b in dummy_replace.items():
-    for k in var_dict.keys():
+    for k in list(var_dict.keys()):
         if a in k:
             var_dict[k.replace(a,b)] = var_dict[k]
             del var_dict[k]
@@ -61,7 +61,7 @@
     ('fc7', 'classifier.3')])
 
 for a, b in layer_map.items():
-    for k in var_dict.keys():
+    for k in list(var_dict.keys()):
         if a in k:
             var_dict[k.replace(a,b)] = var_dict[k]
             del var_dict[k]