From a22929bf5c04fee4934527a5081f5ec7c9f2eb7e Mon Sep 17 00:00:00 2001
From: Mahmoud Ahmed Ali <mahmoudali2929@gmail.com>
Date: Tue, 14 Feb 2023 17:58:15 +0000
Subject: [PATCH] Add instance mask file
---
instance_mask.py | 24 ++++++++++++++++++++++++
1 file changed, 24 insertions(+)
create mode 100644 instance_mask.py
diff --git a/instance_mask.py b/instance_mask.py
new file mode 100644
index 0000000..ead2990
--- /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)
+
--
GitLab