committed by
GitHub
parent
8dd0a2b89c
commit
41b9b92b52
@@ -13,6 +13,7 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import tempfile
|
||||
import unittest
|
||||
|
||||
import numpy as np
|
||||
@@ -298,6 +299,20 @@ class Qwen2VLImageProcessingTest(ImageProcessingTestMixin, unittest.TestCase):
|
||||
expected_output_video_shape = (171500, 1176)
|
||||
self.assertEqual(tuple(encoded_video.shape), expected_output_video_shape)
|
||||
|
||||
def test_custom_image_size(self):
|
||||
for image_processing_class in self.image_processor_list:
|
||||
image_processing = image_processing_class(**self.image_processor_dict)
|
||||
with tempfile.TemporaryDirectory() as tmpdirname:
|
||||
image_processing.save_pretrained(tmpdirname)
|
||||
image_processor_loaded = image_processing_class.from_pretrained(
|
||||
tmpdirname, max_pixels=56 * 56, min_pixels=28 * 28
|
||||
)
|
||||
|
||||
image_inputs = self.image_processor_tester.prepare_image_inputs(equal_resolution=True)
|
||||
prcocess_out = image_processor_loaded(image_inputs, return_tensors="pt")
|
||||
expected_output_video_shape = [112, 1176]
|
||||
self.assertListEqual(list(prcocess_out.pixel_values.shape), expected_output_video_shape)
|
||||
|
||||
@require_vision
|
||||
@require_torch
|
||||
def test_slow_fast_equivalence(self):
|
||||
|
||||
Reference in New Issue
Block a user