From 8510704d9c59aead9db7147fd21d1ffca15da523 Mon Sep 17 00:00:00 2001 From: brothertook Date: Fri, 2 Jul 2021 14:15:13 +0300 Subject: [PATCH] fix parse image size from original PR https://github.com/google/automl/pull/1025 --- efficientdet/utils_test.py | 2 ++ 1 file changed, 2 insertions(+) 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)