ADD file via upload

alphapose/datasets1/layers/PixelUnshuffle.py

@@ -0,0 +1,28 @@
+# -----------------------------------------------------
+# Copyright (c) Shanghai Jiao Tong University. All rights reserved.
+# Written by Jiefeng Li (jeff.lee.sjtu@gmail.com)
+# -----------------------------------------------------
+
+import torch.nn as nn
+
+
+class PixelUnshuffle(nn.Module):
+    '''
+    Initialize: inplanes, planes, upscale_factor
+    OUTPUT: (planes // upscale_factor^2) * ht * wd
+    '''
+
+    def __init__(self, downscale_factor=2):
+        super(PixelUnshuffle, self).__init__()
+        self._r = downscale_factor
+
+    def forward(self, x):
+        b, c, h, w = x.shape
+        out_c = c * (self._r * self._r)
+        out_h = h // self._r
+        out_w = w // self._r
+
+        x_view = x.contiguous().view(b, c, out_h, self._r, out_w, self._r)
+        x_prime = x_view.permute(0, 1, 3, 5, 2, 4).contiguous().view(b, out_c, out_h, out_w)
+
+        return x_prime