"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "dee618a567740e952e55d8b8e1e2b755d4ef89a1"
Unverified Commit c78b0624 authored by vivek rathod's avatar vivek rathod Committed by GitHub
Browse files

Merge pull request #2698 from tombstone/py3compat

fix py3 compatibility issues. 
parents 1e2ada24 730bf06b
...@@ -331,7 +331,7 @@ class ModelBuilderTest(tf.test.TestCase): ...@@ -331,7 +331,7 @@ class ModelBuilderTest(tf.test.TestCase):
}""" }"""
model_proto = model_pb2.DetectionModel() model_proto = model_pb2.DetectionModel()
text_format.Merge(model_text_proto, model_proto) text_format.Merge(model_text_proto, model_proto)
for extractor_type, extractor_class in FEATURE_EXTRACTOR_MAPS.iteritems(): for extractor_type, extractor_class in FEATURE_EXTRACTOR_MAPS.items():
model_proto.faster_rcnn.feature_extractor.type = extractor_type model_proto.faster_rcnn.feature_extractor.type = extractor_type
model = model_builder.build(model_proto, is_training=True) model = model_builder.build(model_proto, is_training=True)
self.assertIsInstance(model, faster_rcnn_meta_arch.FasterRCNNMetaArch) self.assertIsInstance(model, faster_rcnn_meta_arch.FasterRCNNMetaArch)
...@@ -730,7 +730,7 @@ class ModelBuilderTest(tf.test.TestCase): ...@@ -730,7 +730,7 @@ class ModelBuilderTest(tf.test.TestCase):
}""" }"""
model_proto = model_pb2.DetectionModel() model_proto = model_pb2.DetectionModel()
text_format.Merge(model_text_proto, model_proto) text_format.Merge(model_text_proto, model_proto)
for extractor_type, extractor_class in FEATURE_EXTRACTOR_MAPS.iteritems(): for extractor_type, extractor_class in FEATURE_EXTRACTOR_MAPS.items():
model_proto.faster_rcnn.feature_extractor.type = extractor_type model_proto.faster_rcnn.feature_extractor.type = extractor_type
model = model_builder.build(model_proto, is_training=True) model = model_builder.build(model_proto, is_training=True)
self.assertIsInstance(model, rfcn_meta_arch.RFCNMetaArch) self.assertIsInstance(model, rfcn_meta_arch.RFCNMetaArch)
......
...@@ -195,7 +195,7 @@ def merge_external_params_with_configs(configs, hparams=None, **kwargs): ...@@ -195,7 +195,7 @@ def merge_external_params_with_configs(configs, hparams=None, **kwargs):
if hparams: if hparams:
kwargs.update(hparams.values()) kwargs.update(hparams.values())
for key, value in kwargs.iteritems(): for key, value in kwargs.items():
if key == "learning_rate": if key == "learning_rate":
_update_initial_learning_rate(configs, value) _update_initial_learning_rate(configs, value)
tf.logging.info("Overwriting learning rate: %f", value) tf.logging.info("Overwriting learning rate: %f", value)
......
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