Add instance mask file

instance_mask.py

+from matplotlib import pyplot as plt
+import cv2
+
+
+def instance(im, id):
+    im = im * 255
+    im[im == id] = 255
+    im[im < 255] = 0
+    im = im / 255
+    # plt.imshow(im, cmap='gray')
+    # plt.show()
+    return im
+
+
+def generate_instance_mask(data_name):
+    ids = {"Banana": 108, "Orange": 117, "Pear": 121}
+
+    for i in range(4995):
+        for key in ids.keys():
+            id = ids[key]
+            img = plt.imread(f"{data_name}/Mask/{i}.png")
+            instance_img = instance(img, id)
+            cv2.imwrite(f"{data_name}/Instance_Mask/{key}/{i}.png", instance_img)
+