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