TINShift¶
- class mmcv.ops.TINShift[source]¶
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)[source]¶
Perform temporal interlace shift.
- Parameters
input (torch.Tensor) – Feature map with shape [N, num_segments, C, H * W].
shift (torch.Tensor) – Shift tensor with shape [N, num_segments].
- Returns
Feature map after temporal interlace shift.