diff --git a/main.py b/main.py index 05d9dc567ff5e4b8ccdffc6c7d5cefa39d993a60..ab43f89ea2fa850bb1c7dd666b9177be43f4f9b6 100644 --- a/main.py +++ b/main.py @@ -17,11 +17,11 @@ def generate_folders(name, list_categories): is_exist = os.path.exists(name) if not is_exist: os.mkdir(name) - folders = ["RGB", "RGB_Gen", "Meta_Gen", "Depth", "Mask", "Meta", "Pose", "Bbox_2d", "Bbox_2d_loose", "Instance_Segmentation", "Semantic_Segmentation", "Instance_Mask", "Occlusion", "Models", "Pose_transformed", "Bbox", "FPS"] + folders = ["RGB", "RGB_Gen", "RGB_resized", "Meta_Gen", "Depth", "Mask", "Meta", "Pose", "Bbox_2d", "Bbox_2d_loose", "Instance_Segmentation", "Semantic_Segmentation", "Instance_Mask", "Instance_Mask_resized", "Occlusion", "Models", "Pose_transformed", "Bbox", "FPS"] for f in folders: is_exist = os.path.exists(f"{name}/{f}") if not is_exist: - if f not in ["RGB_Gen", "Instance_Mask", "Meta_Gen", "Models", "Pose_transformed", "Bbox", "FPS"]: + if f not in ["RGB_Gen", "RGB_resized", "Instance_Mask", "Instance_Mask_resized", "Meta_Gen", "Models", "Pose_transformed", "Bbox", "FPS"]: os.mkdir(f"{name}/{f}") else: for cat in list_categories: @@ -83,11 +83,13 @@ if __name__ == '__main__': [0.0, 1086.5054444841007, 360.5], [0.0, 0.0, 1.0]]) + new_size = (640, 480) + np.savetxt(f'{dataset_name}/Generated/camera_{choice}.txt', camera) reform_data(dataset_src, dataset_name, dataset_type, Nb_camera, args.World_begin, args.Nb_worlds) - process_compute(dataset_name, camera, Nb_camera, args.World_begin, args.Nb_worlds, list_categories, occ_target, False) + process_compute(dataset_name, camera, new_size, Nb_camera, args.World_begin, args.Nb_worlds, list_categories, occ_target, False) #transform_pose(dataset_name, Nb_camera, Nb_world, list_categories, occ_target) #generate_2d_bbox(dataset_name, Nb_camera, Nb_world, list_categories, occ_target) #generate_instance_mask(dataset_name, Nb_camera, Nb_world, list_categories, occ_target)