From 9f5becc98bb1fd591998c695ed86c8bbade87851 Mon Sep 17 00:00:00 2001 From: vfdev-5 Date: Thu, 14 Jul 2022 17:41:21 +0200 Subject: [PATCH 1/2] [proto] Added missing decorator for center_crop_segmentation_mask tests --- test/test_prototype_transforms_functional.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/test_prototype_transforms_functional.py b/test/test_prototype_transforms_functional.py index c880e8db55b..aaead9bf4ef 100644 --- a/test/test_prototype_transforms_functional.py +++ b/test/test_prototype_transforms_functional.py @@ -495,6 +495,7 @@ def center_crop_bounding_box(): ) +@register_kernel_info_from_sample_inputs_fn def center_crop_segmentation_mask(): for mask, output_size in itertools.product( make_segmentation_masks(image_sizes=((16, 16), (7, 33), (31, 9))), From b2fd4b7415c4adb11c2f94de59a88e199bd3dc17 Mon Sep 17 00:00:00 2001 From: vfdev-5 Date: Thu, 14 Jul 2022 17:44:35 +0200 Subject: [PATCH 2/2] Added center_crop_image_tensor --- test/test_prototype_transforms_functional.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/test/test_prototype_transforms_functional.py b/test/test_prototype_transforms_functional.py index aaead9bf4ef..e369dad6271 100644 --- a/test/test_prototype_transforms_functional.py +++ b/test/test_prototype_transforms_functional.py @@ -487,6 +487,15 @@ def perspective_segmentation_mask(): ) +@register_kernel_info_from_sample_inputs_fn +def center_crop_image_tensor(): + for image, output_size in itertools.product( + make_images(sizes=((16, 16), (7, 33), (31, 9))), + [[4, 3], [42, 70], [4]], # crop sizes < image sizes, crop_sizes > image sizes, single crop size + ): + yield SampleInput(image, output_size) + + @register_kernel_info_from_sample_inputs_fn def center_crop_bounding_box(): for bounding_box, output_size in itertools.product(make_bounding_boxes(), [(24, 12), [16, 18], [46, 48], [12]]):