fix parse image size

from original PR google#1025
brothertook · Jul 2, 2021 · 8510704 · 8510704
1 parent 06d98bc
commit 8510704
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/efficientdet/utils_test.py b/efficientdet/utils_test.py
@@ -67,6 +67,8 @@ def test_image_size(self):
     self.assertEqual(utils.parse_image_size('1280x640'), (640, 1280))
     self.assertEqual(utils.parse_image_size(1280), (1280, 1280))
     self.assertEqual(utils.parse_image_size((1280, 640)), (1280, 640))
+    self.assertEqual(utils.parse_image_size({'width': 640, 'height': 1280}), (1280, 640))
+    self.assertEqual(utils.parse_image_size({'height': 3744, 'width': 5616}), (3744, 5616))
 
   def test_get_feat_sizes(self):
     feats = utils.get_feat_sizes(640, 2)