Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "9e1102990ae91b776b95baabd788e1662d63aa49"
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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
object_detection/utils/variables_helper.py
View file @ 0e363ccf
...@@ -122,7 +122,7 @@ def get_variables_available_in_checkpoint(variables, checkpoint_path): ...@@ -122,7 +122,7 @@ def get_variables_available_in_checkpoint(variables, checkpoint_path):
ckpt_reader = tf.train.NewCheckpointReader(checkpoint_path) ckpt_reader = tf.train.NewCheckpointReader(checkpoint_path)
ckpt_vars = ckpt_reader.get_variable_to_shape_map().keys() ckpt_vars = ckpt_reader.get_variable_to_shape_map().keys()
vars_in_ckpt = {} 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: if variable_name in ckpt_vars:
vars_in_ckpt[variable_name] = variable vars_in_ckpt[variable_name] = variable
else: 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