Skip to content

Commit 8510704

Browse files
authored
fix parse image size
from original PR google#1025
1 parent 06d98bc commit 8510704

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

efficientdet/utils_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,8 @@ def test_image_size(self):
6767
self.assertEqual(utils.parse_image_size('1280x640'), (640, 1280))
6868
self.assertEqual(utils.parse_image_size(1280), (1280, 1280))
6969
self.assertEqual(utils.parse_image_size((1280, 640)), (1280, 640))
70+
self.assertEqual(utils.parse_image_size({'width': 640, 'height': 1280}), (1280, 640))
71+
self.assertEqual(utils.parse_image_size({'height': 3744, 'width': 5616}), (3744, 5616))
7072

7173
def test_get_feat_sizes(self):
7274
feats = utils.get_feat_sizes(640, 2)

0 commit comments

Comments
 (0)