From 18c780bebb8e58d2eece408f4133d314747647d7 Mon Sep 17 00:00:00 2001 From: Nisarg Shah <52718252+nisargshah1999@users.noreply.github.com> Date: Thu, 4 Mar 2021 23:05:00 +0530 Subject: [PATCH] Update g_color.py Earlier error im_base was not defined -- out of scope error during conversion to Image format --- face_parsing/Data_preprocessing/g_color.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/face_parsing/Data_preprocessing/g_color.py b/face_parsing/Data_preprocessing/g_color.py index 4bfb16a..a56787e 100644 --- a/face_parsing/Data_preprocessing/g_color.py +++ b/face_parsing/Data_preprocessing/g_color.py @@ -7,14 +7,17 @@ folder_base = 'CelebAMask-HQ-mask' folder_save = 'CelebAMask-HQ-mask-color' +## Its variable -- Approximate number of Images is 30000 +## so img_num = 30000 img_num = 10 make_folder(folder_save) for k in range(img_num): filename = os.path.join(folder_base, str(k) + '.png') + im_base = np.zeros((512, 512, 3)) if (os.path.exists(filename)): - im_base = np.zeros((512, 512, 3)) + print(filename) im = Image.open(filename) im = np.array(im) for idx, color in enumerate(color_list):