towhee.models.utils.window_reverse3d.window_reverse

towhee.models.utils.window_reverse3d.window_reverse(windows, window_size, b, d, h, w)[source]
Parameters:

  • windows (torch.Tensor) – Tensor with size (B*num_windows, window_size, window_size, C)

  • window_size (tuple[int]) – 3D window size.

  • b (int) – Batch size

  • d (int) – Window size in time dimension.

  • h (int) – Height of image

  • w (int) – Width of image

Returns:

Tensor with size (b, d, h, w, c)

Return type:

x (torch.Tensor)