diff --git a/instance_mask.py b/instance_mask.py
new file mode 100644
index 0000000000000000000000000000000000000000..ead299034d3419bbdd6b77cd3162c25cc3f3c5f4
--- /dev/null
+++ b/instance_mask.py
@@ -0,0 +1,24 @@
+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)
+