From 1b2c67af8a035fa90dc1dc507cdd101df3f5a589 Mon Sep 17 00:00:00 2001 From: Duc Nguyen Date: Sat, 17 Jun 2017 10:14:50 +0900 Subject: [PATCH] Fix compatibility for model_builder_test.py (#1571) iteritems -> items --- object_detection/builders/model_builder_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/object_detection/builders/model_builder_test.py b/object_detection/builders/model_builder_test.py index 513c5fab359..28b15e6508f 100644 --- 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)