From a2e3e4d639dca7842fb407ff33f89213d69bc19c Mon Sep 17 00:00:00 2001 From: Genk Date: Wed, 20 Mar 2019 11:26:03 +0800 Subject: [PATCH] Add files via upload Please use this code for testing your trained method --- Training_Code/Processing_Im.m | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 Training_Code/Processing_Im.m diff --git a/Training_Code/Processing_Im.m b/Training_Code/Processing_Im.m new file mode 100644 index 0000000..3beb871 --- /dev/null +++ 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 +