test

Author: LinkWoong

LinkGAN/essential.py

@@ -92,7 +92,9 @@ def show_active_img_and_save_denoise_filter(name,image,path):
     return mat
 
 
-def show_active_img_and_save_denoise_filter2(name,image,path):
+def show_active_img_and_save_denoise_filter2(name,image,path, i):
+
+    path = path + '/sketch'
     mat = image.astype(np.float)
     mat[mat<0.1] = 0
     mat = -mat + 1
@@ -104,7 +106,7 @@ def show_active_img_and_save_denoise_filter2(name,image,path):
     mat = ndimage.median_filter(mat,1)
     if not os.path.exists(path):
         os.makedirs(path)
-    
+    path = path + '/sketched_' + str(i) + '.jpeg'
     cv2.imwrite(path,mat)
 
     return mat

LinkGAN/preprocess.py

@@ -54,10 +54,10 @@ def preprocess(image_path, mod_path, content_to_sketch):
 			show_active_img_and_save('sketches', line_mat, './' + str(i) + '.jpeg')
 			line_mat = np.amax(line_mat, 2)
 
-			show_active_img_and_save_denoise_filter2('sketches', line_mat, new_path + '/sketch/sketched_' + str(i) + '.jpeg')
+			show_active_img_and_save_denoise_filter2('sketches', line_mat, new_path, i)
 			show_active_img_and_save_denoise_filter('sketches', line_mat, './' + str(i) + '.jpeg')
 
 			sketch_path = new_path + '/sketch'
 			content_path = new_path
 
-			return content_path, sketch_path
+	return content_path, sketch_path

LinkGAN/test.py

@@ -0,0 +1,22 @@
+import tensorflow as tf
+import numpy as np
+import io
+import os
+os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
+import cv2
+import pandas as pd 
+
+
+from keras.models import load_model
+from preprocess import *
+
+path = '/media/linkwong/D/1girl'
+mod_path = '/media/linkwong/D/mod.h5'
+
+content_to_sketch = True
+
+
+content_path, sketch_path = preprocess(path, mod_path, content_to_sketch)
+
+print content_path
+print sketch_path