Upload create_mask.py

create_mask.py

@@ -0,0 +1,118 @@
+# Author: aqeelanwar
+# Created: 6 July,2020, 12:14 AM
+# Email: aqeel.anwar@gatech.edu
+
+from PIL import ImageColor
+import cv2
+import numpy as np
+
+COLOR = [
+    "#fc1c1a",
+    "#177ABC",
+    "#94B6D2",
+    "#A5AB81",
+    "#DD8047",
+    "#6b425e",
+    "#e26d5a",
+    "#c92c48",
+    "#6a506d",
+    "#ffc900",
+    "#ffffff",
+    "#000000",
+    "#49ff00",
+]
+
+
+def color_the_mask(mask_image, color, intensity):
+    assert 0 <= intensity <= 1, "intensity should be between 0 and 1"
+    RGB_color = ImageColor.getcolor(color, "RGB")
+    RGB_color = (RGB_color[2], RGB_color[1], RGB_color[0])
+    orig_shape = mask_image.shape
+    bit_mask = mask_image[:, :, 3]
+    mask_image = mask_image[:, :, 0:3]
+
+    color_image = np.full(mask_image.shape, RGB_color, np.uint8)
+    mask_color = cv2.addWeighted(mask_image, 1 - intensity, color_image, intensity, 0)
+    mask_color = cv2.bitwise_and(mask_color, mask_color, mask=bit_mask)
+    colored_mask = np.zeros(orig_shape, dtype=np.uint8)
+    colored_mask[:, :, 0:3] = mask_color
+    colored_mask[:, :, 3] = bit_mask
+    return colored_mask
+
+
+def texture_the_mask(mask_image, texture_path, intensity):
+    assert 0 <= intensity <= 1, "intensity should be between 0 and 1"
+    orig_shape = mask_image.shape
+    bit_mask = mask_image[:, :, 3]
+    mask_image = mask_image[:, :, 0:3]
+    texture_image = cv2.imread(texture_path)
+    texture_image = cv2.resize(texture_image, (orig_shape[1], orig_shape[0]))
+
+    mask_texture = cv2.addWeighted(
+        mask_image, 1 - intensity, texture_image, intensity, 0
+    )
+    mask_texture = cv2.bitwise_and(mask_texture, mask_texture, mask=bit_mask)
+    textured_mask = np.zeros(orig_shape, dtype=np.uint8)
+    textured_mask[:, :, 0:3] = mask_texture
+    textured_mask[:, :, 3] = bit_mask
+
+    return textured_mask
+
+
+
+# cloth_mask = cv2.imread("masks/templates/cloth.png", cv2.IMREAD_UNCHANGED)
+# # cloth_mask = color_the_mask(cloth_mask, color=COLOR[0], intensity=0.5)
+# path = "masks/textures"
+# path, dir, files = os.walk(path).__next__()
+# first_frame = True
+# col_limit = 6
+# i = 0
+# # img_concat_row=[]
+# img_concat = []
+# # for f in files:
+# #     if "._" not in f:
+# #         print(f)
+# #         i += 1
+# #         texture_image = cv2.imread(os.path.join(path, f))
+# #         m = texture_the_mask(cloth_mask, texture_image, intensity=0.5)
+# #         if first_frame:
+# #             img_concat_row = m
+# #             first_frame = False
+# #         else:
+# #             img_concat_row = cv2.hconcat((img_concat_row, m))
+# #
+# #             if i % col_limit == 0:
+# #                 if len(img_concat) > 0:
+# #                     img_concat = cv2.vconcat((img_concat, img_concat_row))
+# #                 else:
+# #                     img_concat = img_concat_row
+# #                 first_frame = True
+#
+# ## COlor the mask
+# thresholds = np.arange(0.1,0.9,0.05)
+# for intensity in thresholds:
+#     c=COLOR[2]
+#     # intensity = 0.5
+#     if "._" not in c:
+#         print(intensity)
+#         i += 1
+#         # texture_image = cv2.imread(os.path.join(path, f))
+#         m = color_the_mask(cloth_mask, c, intensity=intensity)
+#         if first_frame:
+#             img_concat_row = m
+#             first_frame = False
+#         else:
+#             img_concat_row = cv2.hconcat((img_concat_row, m))
+#
+#             if i % col_limit == 0:
+#                 if len(img_concat) > 0:
+#                     img_concat = cv2.vconcat((img_concat, img_concat_row))
+#                 else:
+#                     img_concat = img_concat_row
+#                 first_frame = True
+#
+#
+# cv2.imshow("k", img_concat)
+# cv2.imwrite("combine_N95_left.png", img_concat)
+# cv2.waitKey(0)
+# cc = 1