Update src/transformers/models/rt_detr/image_processing_rt_detr.py

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
This commit is contained in:
Sangbum Daniel Choi 2024-04-12 21:52:32 +09:00 committed by GitHub
parent c6c62b28fb
commit 74aa3e1de0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1046,7 +1046,7 @@ class RTDetrImageProcessor(BaseImageProcessor):
]
if do_rescale:
images = [self.rescale(image, rescale_factor, input_data_format=input_data_format) for image in images]
images = [self.rescale(image, scale=rescale_factor, input_data_format=input_data_format) for image in images]
if do_normalize:
images = [