Fix compatibility for model_builder_test.py (tensorflow#1571)

iteritems -> items
envin3 · Jun 17, 2017 · 1b2c67a · 1b2c67a
1 parent 8d91ce7
commit 1b2c67a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/object_detection/builders/model_builder_test.py b/object_detection/builders/model_builder_test.py
@@ -255,7 +255,7 @@ def test_create_faster_rcnn_resnet_v1_models_from_config(self):
       }"""
     model_proto = model_pb2.DetectionModel()
     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 = model_builder.build(model_proto, is_training=True)
       self.assertIsInstance(model, faster_rcnn_meta_arch.FasterRCNNMetaArch)
@@ -445,7 +445,7 @@ def test_create_rfcn_resnet_v1_model_from_config(self):
       }"""
     model_proto = model_pb2.DetectionModel()
     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 = model_builder.build(model_proto, is_training=True)
       self.assertIsInstance(model, rfcn_meta_arch.RFCNMetaArch)