towhee.models.utils.get_window_size.get_window_size¶
- towhee.models.utils.get_window_size.get_window_size(x_size, window_size, shift_size=None)[source]¶
- Parameters:
x_size (tuple[int]) – Tensor with size (B*num_windows, window_size, window_size, C)
window_size (tuple[int]) – 3D window size.
shift_size (tuple[int]) – Shift size. ref: https://arxiv.org/pdf/2106.13230.pdf
- Returns (tuple[int]):
shifted window size.