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)