Unverified Commit 21a2fbaf authored by Wesley L Passos's avatar Wesley L Passos Committed by GitHub
Browse files

Fix docstring in get_oneformer_resize_output_image_size func (#27207)

parent f8afb2b2
......@@ -307,14 +307,16 @@ def get_oneformer_resize_output_image_size(
Computes the output size given the desired size.
Args:
input_image (`np.ndarray`):
image (`np.ndarray`):
The input image.
size (`int`, `Tuple[int, int]`, `List[int]`, `Tuple[int]`):
size (`int` or `Tuple[int, int]` or `List[int]` or `Tuple[int]`):
The size of the output image.
max_size (`int`, *optional*):
The maximum size of the output image.
default_to_square (`bool`, *optional*, defaults to `True`):
Whether to default to square if no size is provided.
input_data_format (`ChannelDimension` or `str`, *optional*):
The channel dimension format of the input image. If unset, will use the inferred format from the input.
Returns:
`Tuple[int, int]`: The output size.
......
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