TINShift¶
- class mmcv.ops.TINShift[源代码]¶
Temporal Interlace Shift.
Temporal Interlace shift is a differentiable temporal-wise frame shifting which is proposed in “Temporal Interlacing Network”
Please refer to Temporal Interlacing Network for more details.
Code is modified from https://github.com/mit-han-lab/temporal-shift-module
- forward(input, shift)[源代码]¶
Perform temporal interlace shift.
- 参数
input (torch.Tensor) – Feature map with shape [N, num_segments, C, H * W].
shift (torch.Tensor) – Shift tensor with shape [N, num_segments].
- 返回
Feature map after temporal interlace shift.