pull/1178/head
gaotongxiao 2022-06-28 19:21:56 +08:00
parent 24575de140
commit 0fb0d7cb1a
1 changed files with 2 additions and 2 deletions

View File

@ -477,7 +477,7 @@ class TestEastRandomCrop(unittest.TestCase):
resize_results = resize(crop_results)
self.assertEqual(resize_results['img'].shape, (15, 30, 3))
self.assertEqual(crop_results['img_shape'], (15, 30))
self.assertEqual(crop_results['scale'], (30, 15))
self.assertEqual(crop_results['scale'], (30, 30))
self.assertEqual(crop_results['scale_factor'], (1., 1.))
self.assertTrue(crop_results['keep_ratio'])
@ -513,7 +513,7 @@ class TestRandomResize(unittest.TestCase):
self.assertEqual(results['img'].shape, (450, 600, 3))
self.assertEqual(results['img_shape'], (450, 600))
self.assertEqual(results['keep_ratio'], True)
self.assertEqual(results['scale'], (600, 450))
self.assertEqual(results['scale'], (600, 600))
self.assertEqual(results['scale_factor'], (600. / 400., 450. / 300.))
self.assertTrue(