🚨 Update image_processing_vitmatte.py (#30566)

* Update image_processing_vitmatte.py * add test * [run-slow]vitmatte
2024-05-02 12:00:07 +02:00
parent 12c5544dca
commit f95302584b
2 changed files with 7 additions and 3 deletions
--- a/tests/models/vitmatte/test_image_processing_vitmatte.py
+++ b/tests/models/vitmatte/test_image_processing_vitmatte.py
@@ -192,3 +192,7 @@ class VitMatteImageProcessingTest(ImageProcessingTestMixin, unittest.TestCase):
        image = np.random.randn(3, 249, 491)
        images = image_processing.pad_image(image)
        assert images.shape == (3, 256, 512)
+
+        image = np.random.randn(3, 249, 512)
+        images = image_processing.pad_image(image)
+        assert images.shape == (3, 256, 512)