Skip to content

GitLab

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

YOLOS - Revert default return_pixel_mask value (#25404) 

Revert default return_pixel_mask value
parent 59937716
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
src/transformers/models/yolos/image_processing_yolos.py
View file @ ea5dda22
...@@ -892,12 +892,11 @@ class YolosImageProcessor(BaseImageProcessor): ...@@ -892,12 +892,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,
) -> BatchFeature: ) -> BatchFeature:
......
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