From f7f292833d313f146520b77441f74e1c689cccd7 Mon Sep 17 00:00:00 2001
From: cszn <cskaizhang@gmail.com>
Date: Sat, 23 Dec 2017 17:21:41 +0800
Subject: [PATCH] Init

---
 utilities/vl_nnSubP.m | 78 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100644 utilities/vl_nnSubP.m

diff --git a/utilities/vl_nnSubP.m b/utilities/vl_nnSubP.m
new file mode 100644
index 0000000..acdc9bc
--- /dev/null
+++ b/utilities/vl_nnSubP.m
@@ -0,0 +1,78 @@
+function y = vl_nnSubP(x, dzdy, varargin)
+
+
+opts.scale = 2;
+opts = vl_argparse(opts, varargin, 'nonrecursive') ;
+scale = opts.scale;
+
+if scale > 1 % many (small)---> one (big) e.g., 100X100X256X128  ----> 200X200X64*128
+    scale2 = scale^2;
+    if nargin <= 1 || isempty(dzdy)
+        [hei,wid,channelI,bsize] = size(x);
+        channelO = channelI/scale2;
+        channelI_start = 0;
+        idx1 = scale * (1:hei);
+        idx2 = scale * (1:wid);
+        y = zeros([scale*hei, scale*wid, channelO, bsize],'like',x);
+        for nchannelO = 1:channelO
+            for i = 1:scale2
+                a = mod(i-1, scale) + 1;
+                b = floor((i-1)/scale) + 1;
+                y(idx1+a-scale, idx2+b-scale, nchannelO, :) = x(:,:,channelI_start+i,:);
+            end
+            channelI_start = scale2 + channelI_start;
+        end
+    else
+        [hei,wid,channelO,bsize] = size(dzdy);
+        idx1 = 1:scale:hei;
+        idx2 = 1:scale:wid;
+        channelI = channelO*scale2;
+        channelI_start = 0;
+        y = zeros([hei/scale, wid/scale, channelI, bsize],'like',x);
+        for  nchannelO = 1:channelO
+            for i = 1:scale^2
+                a = mod(i-1, scale) + 1;
+                b = floor((i-1)/scale) + 1;
+                y(:, :, channelI_start+i, :) = dzdy(idx1+a-1, idx2+b-1,nchannelO,:);
+            end
+            channelI_start = scale2 + channelI_start;
+        end
+    end
+    
+else %  one (big) ---> many (small)  e.g., 200X200X64*128--->100X100X256X128
+    
+    scale = round(1/scale);
+    scale2 = scale^2;
+    if nargin <= 1 || isempty(dzdy)
+        [hei,wid,channelO,bsize] = size(x);
+        idx1 = 1:scale:hei;
+        idx2 = 1:scale:wid;
+        channelI = channelO*scale2;
+        channelI_start = 0;
+        y = zeros([hei/scale, wid/scale, channelI, bsize],'like',x);
+        for  nchannelO = 1:channelO
+            for i = 1:scale^2
+                a = mod(i-1, scale) + 1;
+                b = floor((i-1)/scale) + 1;
+                y(:, :, channelI_start+i, :) = x(idx1+a-1, idx2+b-1,nchannelO,:);
+            end
+            channelI_start = scale2 + channelI_start;
+        end
+    else
+        [hei,wid,channelI,bsize] = size(dzdy);
+        channelO = channelI/scale2;
+        channelI_start = 0;
+        idx1 = scale * (1:hei);
+        idx2 = scale * (1:wid);
+        y = zeros([scale*hei, scale*wid, channelO, bsize],'like',x);
+        for nchannelO = 1:channelO
+            for i = 1:scale2
+                a = mod(i-1, scale) + 1;
+                b = floor((i-1)/scale) + 1;
+                y(idx1+a-scale, idx2+b-scale, nchannelO, :) = dzdy(:,:,channelI_start+i,:);
+            end
+            channelI_start = scale2 + channelI_start;
+        end
+    end
+end
+