diff --git a/filt_cnn_artifact.m b/filt_cnn_artifact.m index 815c53d..28732fb 100644 --- a/filt_cnn_artifact.m +++ b/filt_cnn_artifact.m @@ -1,24 +1,26 @@ -% parpool(8) -% parfor i = 0 : 43 -for i = 2:2 % 70:72 -% parfor i = 0 : 149 + +for i = 0:34 close all + + in_fn = ['results/' int2str(i) '_final_res.png']; + out_fn = ['results/' int2str(i) '_final_res2.png']; - in_fn = [int2str(i) '_result.jpg']; - out_fn = [int2str(i) '_result2.jpg']; - - -% in_fn = ['../tmp36_wikiart2_current_more/' int2str(i) '_result.png']; -% out_fn = ['../tmp36_wikiart2_current_more/' int2str(i) '_result2.png']; - - + if exist(in_fn, 'file') ~= 2 + fprintf('file doesn''t exist: %s\n', in_fn); + continue + end + if exist(out_fn, 'file') == 2 + fprintf('result already exists: %s\n', out_fn); + continue + end + I = im2double(imread(in_fn)); - - G = im2double(imread(['free2use/' int2str(i) '_naive.png'])); - M = im2double(imread(['free2use/' int2str(i) '_c_mask.png'])); - B = im2double(imread(['free2use/' int2str(i) '_target.jpg'])); - - + + G = im2double(imread(['data/' int2str(i) '_naive.jpg'])); + M = im2double(imread(['data/' int2str(i) '_c_mask.jpg'])); + B = im2double(imread(['data/' int2str(i) '_target.jpg'])); + + tr= 3; h = fspecial('gaussian', [2*tr+1 2*tr+1], tr); sM = imfilter(M, h, 'same'); @@ -26,11 +28,11 @@ sM(sM < 0.01) = 0; sM = imfilter(sM, h, 'same'); - addpath code/colorspace + addpath 3rdparty/colorspace I_lab = colorspace('rgb->lab', I); - addpath code/guided_filter - addpath code/patchmatch-2.0 + addpath 3rdparty/guided_filter + addpath 3rdparty/patchmatch-2.0 r = 2; % try r=2, 4, or 8 eps = 0.1^2; % try eps=0.1^2, 0.2^2, 0.4^2 @@ -47,12 +49,12 @@ patch_w = 7; ann = nnmex(O1, B, algo, patch_w, [], [], [], [], [], cores); O2_base = im2double(votemex(B, ann)); - + r = 3; h = fspecial('gaussian', [2*r+1 2*r+1], r/3); O1_base = imfilter(O1, h, 'same'); O2 = O2_base + O1 - O1_base; - + O2 = O2.*sM + B.*(1-sM); figure; imshow(I) figure; imshow(O2) diff --git a/results/0_final_res2.png b/results/0_final_res2.png new file mode 100644 index 0000000..9d71aee Binary files /dev/null and b/results/0_final_res2.png differ diff --git a/results/10_final_res2.png b/results/10_final_res2.png new file mode 100644 index 0000000..87147f9 Binary files /dev/null and b/results/10_final_res2.png differ diff --git a/results/11_final_res2.png b/results/11_final_res2.png new file mode 100644 index 0000000..b375ec4 Binary files /dev/null and b/results/11_final_res2.png differ diff --git a/results/12_final_res2.png b/results/12_final_res2.png new file mode 100644 index 0000000..5a15684 Binary files /dev/null and b/results/12_final_res2.png differ diff --git a/results/13_final_res2.png b/results/13_final_res2.png new file mode 100644 index 0000000..849977f Binary files /dev/null and b/results/13_final_res2.png differ diff --git a/results/14_final_res2.png b/results/14_final_res2.png new file mode 100644 index 0000000..7ed4fb1 Binary files /dev/null and b/results/14_final_res2.png differ diff --git a/results/15_final_res2.png b/results/15_final_res2.png new file mode 100644 index 0000000..a1cb510 Binary files /dev/null and b/results/15_final_res2.png differ diff --git a/results/16_final_res2.png b/results/16_final_res2.png new file mode 100644 index 0000000..be5fcd1 Binary files /dev/null and b/results/16_final_res2.png differ diff --git a/results/17_final_res2.png b/results/17_final_res2.png new file mode 100644 index 0000000..4f58a69 Binary files /dev/null and b/results/17_final_res2.png differ diff --git a/results/18_final_res2.png b/results/18_final_res2.png new file mode 100644 index 0000000..ce31c0d Binary files /dev/null and b/results/18_final_res2.png differ diff --git a/results/19_final_res2.png b/results/19_final_res2.png new file mode 100644 index 0000000..8777a4e Binary files /dev/null and b/results/19_final_res2.png differ diff --git a/results/1_final_res2.png b/results/1_final_res2.png new file mode 100644 index 0000000..7d65454 Binary files /dev/null and b/results/1_final_res2.png differ diff --git a/results/20_final_res2.png b/results/20_final_res2.png new file mode 100644 index 0000000..025597a Binary files /dev/null and b/results/20_final_res2.png differ diff --git a/results/21_final_res2.png b/results/21_final_res2.png new file mode 100644 index 0000000..9a919c1 Binary files /dev/null and b/results/21_final_res2.png differ diff --git a/results/22_final_res2.png b/results/22_final_res2.png new file mode 100644 index 0000000..5e4f45d Binary files /dev/null and b/results/22_final_res2.png differ diff --git a/results/23_final_res2.png b/results/23_final_res2.png new file mode 100644 index 0000000..4dbd59d Binary files /dev/null and b/results/23_final_res2.png differ diff --git a/results/24_final_res2.png b/results/24_final_res2.png new file mode 100644 index 0000000..32c40d6 Binary files /dev/null and b/results/24_final_res2.png differ diff --git a/results/25_final_res2.png b/results/25_final_res2.png new file mode 100644 index 0000000..21d887b Binary files /dev/null and b/results/25_final_res2.png differ diff --git a/results/26_final_res2.png b/results/26_final_res2.png new file mode 100644 index 0000000..91f6372 Binary files /dev/null and b/results/26_final_res2.png differ diff --git a/results/27_final_res2.png b/results/27_final_res2.png new file mode 100644 index 0000000..6909de8 Binary files /dev/null and b/results/27_final_res2.png differ diff --git a/results/28_final_res2.png b/results/28_final_res2.png new file mode 100644 index 0000000..95381c6 Binary files /dev/null and b/results/28_final_res2.png differ diff --git a/results/29_final_res2.png b/results/29_final_res2.png new file mode 100644 index 0000000..56443c0 Binary files /dev/null and b/results/29_final_res2.png differ diff --git a/results/2_final_res2.png b/results/2_final_res2.png new file mode 100644 index 0000000..d7bd667 Binary files /dev/null and b/results/2_final_res2.png differ diff --git a/results/30_final_res2.png b/results/30_final_res2.png new file mode 100644 index 0000000..e238bde Binary files /dev/null and b/results/30_final_res2.png differ diff --git a/results/31_final_res2.png b/results/31_final_res2.png new file mode 100644 index 0000000..fa0efd4 Binary files /dev/null and b/results/31_final_res2.png differ diff --git a/results/32_final_res2.png b/results/32_final_res2.png new file mode 100644 index 0000000..59a2cb3 Binary files /dev/null and b/results/32_final_res2.png differ diff --git a/results/33_final_res2.png b/results/33_final_res2.png new file mode 100644 index 0000000..11a894f Binary files /dev/null and b/results/33_final_res2.png differ diff --git a/results/34_final_res2.png b/results/34_final_res2.png new file mode 100644 index 0000000..858e772 Binary files /dev/null and b/results/34_final_res2.png differ diff --git a/results/3_final_res2.png b/results/3_final_res2.png new file mode 100644 index 0000000..988e392 Binary files /dev/null and b/results/3_final_res2.png differ diff --git a/results/4_final_res2.png b/results/4_final_res2.png new file mode 100644 index 0000000..d110543 Binary files /dev/null and b/results/4_final_res2.png differ diff --git a/results/5_final_res2.png b/results/5_final_res2.png new file mode 100644 index 0000000..79b9794 Binary files /dev/null and b/results/5_final_res2.png differ diff --git a/results/6_final_res2.png b/results/6_final_res2.png new file mode 100644 index 0000000..e45aca1 Binary files /dev/null and b/results/6_final_res2.png differ diff --git a/results/7_final_res2.png b/results/7_final_res2.png new file mode 100644 index 0000000..2bad652 Binary files /dev/null and b/results/7_final_res2.png differ diff --git a/results/8_final_res2.png b/results/8_final_res2.png new file mode 100644 index 0000000..fee1912 Binary files /dev/null and b/results/8_final_res2.png differ diff --git a/results/9_final_res2.png b/results/9_final_res2.png new file mode 100644 index 0000000..b4e8a81 Binary files /dev/null and b/results/9_final_res2.png differ