Commit 0e363ccf authored by derekjchow's avatar derekjchow Committed by GitHub
Browse files

Merge pull request #1763 from ByeongjooAhn/bahn-objectdetection-py3

Fix compatibility for Python3 (object_detection/utils/variables_helper.py)
parents 58417e76 8f16def1
......@@ -122,7 +122,7 @@ def get_variables_available_in_checkpoint(variables, checkpoint_path):
ckpt_reader = tf.train.NewCheckpointReader(checkpoint_path)
ckpt_vars = ckpt_reader.get_variable_to_shape_map().keys()
vars_in_ckpt = {}
for variable_name, variable in sorted(variable_names_map.iteritems()):
for variable_name, variable in sorted(variable_names_map.items()):
if variable_name in ckpt_vars:
vars_in_ckpt[variable_name] = variable
else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment