diff --git a/efficientdet/utils_test.py b/efficientdet/utils_test.py
index 7ca479e80..e733d1178 100644
--- 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)