zhangyue66 commited on
Commit ·
264a95e
1
Parent(s): 143b689
update
Browse files
image_processing_paddleocr_vl.py
CHANGED
|
@@ -339,6 +339,7 @@ class PaddleOCRVLImageProcessor(BaseImageProcessor):
|
|
| 339 |
images = make_list_of_images(images)
|
| 340 |
|
| 341 |
if input_data_format is None:
|
|
|
|
| 342 |
input_data_format = ChannelDimension.LAST if isinstance(images[0], Image.Image) else infer_channel_dimension_format(images[0])
|
| 343 |
|
| 344 |
if do_convert_rgb:
|
|
|
|
| 339 |
images = make_list_of_images(images)
|
| 340 |
|
| 341 |
if input_data_format is None:
|
| 342 |
+
# We assume that all images have the same channel dimension format.
|
| 343 |
input_data_format = ChannelDimension.LAST if isinstance(images[0], Image.Image) else infer_channel_dimension_format(images[0])
|
| 344 |
|
| 345 |
if do_convert_rgb:
|