Skip to content

GitLab

Unverified Commit e50c9253 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

YOLOS - reset default return_pixel_mask value (#25559) 

Remove added back copied from statement
parent c8346cb2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
src/transformers/models/yolos/image_processing_yolos.py
View file @ e50c9253
...@@ -952,12 +952,11 @@ class YolosImageProcessor(BaseImageProcessor): ...@@ -952,12 +952,11 @@ class YolosImageProcessor(BaseImageProcessor):
) )
return padded_image return padded_image
# Copied from transformers.models.detr.image_processing_detr.DetrImageProcessor.pad
def pad( def pad(
self, self,
images: List[np.ndarray], images: List[np.ndarray],
constant_values: Union[float, Iterable[float]] = 0, constant_values: Union[float, Iterable[float]] = 0,
return_pixel_mask: bool = True, return_pixel_mask: bool = False,
return_tensors: Optional[Union[str, TensorType]] = None, return_tensors: Optional[Union[str, TensorType]] = None,
data_format: Optional[ChannelDimension] = None, data_format: Optional[ChannelDimension] = None,
input_data_format: Optional[Union[str, ChannelDimension]] = None, input_data_format: Optional[Union[str, ChannelDimension]] = None,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment