Skip to content

GitLab

"x/vscode:/vscode.git/clone" did not exist on "33ee7168ba1e16c813b52dc2c9417efa1e2e9f20"
Commit 1b2c67af authored by Duc Nguyen's avatar Duc Nguyen Committed by Sergio Guadarrama
Browse files

Fix compatibility for model_builder_test.py (#1571) 

iteritems -> items
parent 8d91ce76
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
object_detection/builders/model_builder_test.py
View file @ 1b2c67af
...@@ -255,7 +255,7 @@ class ModelBuilderTest(tf.test.TestCase): ...@@ -255,7 +255,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)
...@@ -445,7 +445,7 @@ class ModelBuilderTest(tf.test.TestCase): ...@@ -445,7 +445,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)
......
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