Add files via upload

Please use this code for testing your trained method
lpj0 · Mar 20, 2019 · a2e3e4d · a2e3e4d
1 parent cdb8fdc
commit a2e3e4d
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/Training_Code/Processing_Im.m b/Training_Code/Processing_Im.m
@@ -0,0 +1,16 @@
+function output1 = Processing_Im(im, net, gpu, out_idx)
+%     tmp = net.params(48).value;
+%     net.params(48).value = net.params(68).value;
+%     net.params(68).value = tmp;
+
+    input = im2single(im(:,:,1));
+    if gpu
+        %input = gpuArray(input*4-2);
+		input = gpuArray(input);
+    end
+    net.eval({'input',input}) ;
+    %%% output (single)
+    output1 = gather(squeeze(gather(net.vars(out_idx).value+2)/4));
+    output1 = im2uint8(output1);
+end
+