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.