towhee.models.utils.window_partition.window_partition

towhee.models.utils.window_partition.window_partition(x, window_size: int)[source]
Parameters:

  • x – (b, h, w, c)

  • window_size (int) – window size

Returns:

(num_windows*B, window_size, window_size, c)

Return type:

windows