From 6887713edb0d7759c78c148b836e3cfba6c2991c Mon Sep 17 00:00:00 2001
From: Gduret <guillaume.duret@ec-lyon.fr>
Date: Wed, 1 Mar 2023 13:46:32 +0100
Subject: [PATCH] resize rgb and instance

---
 resize.py | 64 +++++++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 50 insertions(+), 14 deletions(-)

diff --git a/resize.py b/resize.py
index 70cc608..6174f8c 100644
--- a/resize.py
+++ b/resize.py
@@ -1,12 +1,26 @@
 from matplotlib import image
 import matplotlib.pyplot as plt
+from utils import compute_categories_id, compute_id_good_occ
 import cv2
 import numpy as np
-
+import argparse
 
 if __name__ == '__main__':
 
+
+
+    # Create the parser
+    parser = argparse.ArgumentParser()
+    # Add an argument
+    parser.add_argument('--Nb_worlds', type=int, required=True)
+    parser.add_argument('--World_begin', type=int, required=True)
+    # Parse the argument
+    args = parser.parse_args()
+
+
     choice = "low"
+    Nb_instance = 1
+    occ_target = 0.5
 
     if choice == 'high':
         camera = np.matrix([[1386.4138492513919, 0.0, 960.5],
@@ -28,27 +42,49 @@ if __name__ == '__main__':
 
     dataset_name = f"GUIMOD_{choice}"
     new_size = (640, 480)
+    Nb_camera = 15
+    list_categories = ["banana1", "kiwi1", "pear2", "strawberry1", "apricot", "orange2", "peach1", "lemon2", "apple2" ]
+
+    for i in range(args.World_begin, args.World_begin + args.Nb_worlds): # worlds
+        
+        catergories_instance_array_id_to_cat, catergories_instance_array_cat_to_id, catergories_label_to_id = compute_categories_id(dataset_name, i)
+        
+        for j in range(1, Nb_camera+1): # cameras
+            p = ((i-1)*Nb_camera) + j
+
+            catergories_occ_array = compute_id_good_occ(dataset_name, p, catergories_instance_array_id_to_cat, catergories_instance_array_cat_to_id, occ_target)
+
+            #depth = cv2.resize(cv2.imread(f"{dataset_name}/Depth/{i}.tiff"), new_size)
+
+            for categories in list_categories:
+                if categories in catergories_occ_array.keys():
+                    if len(catergories_occ_array[categories]) == 1 :
+
+                        print(f"{dataset_name}/Generated/{categories}/RGB_Gen/{p}.png")
+                        rgb = cv2.resize(cv2.imread(f"{dataset_name}/Generated/{categories}/RGB_Gen/{p}.png"), new_size)
+                        cv2.imwrite(f"{dataset_name}/Generated/{categories}/RGB_resized/{p}.png", rgb)
 
-    for i in range(4995):
-        rgb = cv2.resize(cv2.imread(f"{dataset_name}/RGB/{i}.png"), new_size)
-        cv2.imwrite(f"rgb/{i}.png", rgb)
+                        #mask = cv2.resize(cv2.imread(f"{dataset_name}//Instance_Mask/{i}.png"), new_size)
+                        #cv2.imwrite(f"mask/{i}.png", mask*255)
 
-        mask = cv2.resize(cv2.imread(f"{dataset_name}/Mask/{i}.png"), new_size)
-        cv2.imwrite(f"mask/{i}.png", mask*255)
+                        print(f"{dataset_name}/Generated/{categories}/Instance_Mask/{p}.png")
+                        print("new_size",new_size)
+                        cat_mask = cv2.resize(cv2.imread(f"{dataset_name}/Generated/{categories}/Instance_Mask/{p}.png"), new_size)
 
-        depth = cv2.resize(cv2.imread(f"{dataset_name}/Depth/{i}.tiff"), new_size)
+                        print(f"{dataset_name}/Generated/{categories}/Instance_Mask_resized/{p}.png")
+                        cv2.imwrite(f"{dataset_name}/Generated/{categories}/Instance_Mask_resized/{p}.png", cat_mask)
 
-        banana_mask = cv2.resize(cv2.imread(f"{dataset_name}/Instance_Mask/banana1/{i}.png"), new_size)
-        cv2.imwrite(f"banana1_mask/{i}.png", banana_mask*255)
+                        # banana_mask = cv2.resize(cv2.imread(f"{dataset_name}/Instance_Mask/banana1/{i}.png"), new_size)
+                        # cv2.imwrite(f"banana1_mask/{i}.png", banana_mask*255)
 
-        orange_mask = cv2.resize(cv2.imread(f"{dataset_name}/Instance_Mask/orange2/{i}.png"), new_size)
-        cv2.imwrite(f"orange2_mask/{i}.png", orange_mask*255)
+                        # orange_mask = cv2.resize(cv2.imread(f"{dataset_name}/Instance_Mask/orange2/{i}.png"), new_size)
+                        # cv2.imwrite(f"orange2_mask/{i}.png", orange_mask*255)
 
-        Pear_mask = cv2.resize(cv2.imread(f"{dataset_name}/Instance_Mask/pear2/{i}.png"), new_size)
-        cv2.imwrite(f"pear2_mask/{i}.png", Pear_mask*255)
+                        # Pear_mask = cv2.resize(cv2.imread(f"{dataset_name}/Instance_Mask/pear2/{i}.png"), new_size)
+                        # cv2.imwrite(f"pear2_mask/{i}.png", Pear_mask*255)
 
 
-        print("Done")
+            print("Done")
 
         # Check 2d bbox to resize it
 
-- 
GitLab