Testing how to make data generator application

2D_Data_Generator_Model.py

@@ -17,6 +17,8 @@ from Data_Generation.Shape_Generation_Functions import basic_box, diagonal_box_s
     back_slash_box, forward_slash_box, back_slash_plus_box, forward_slash_plus_box, hot_dog_box, hamburger_box, \
     x_hamburger_box, x_hot_dog_box, x_plus_box
 
+
+
 from Data_Generation.Dataset_Generation_Functions import make_boxes
 
 
@@ -28,8 +30,8 @@ food = load_dataset("cmudrc/2d-lattices", split="train+test")  # Loads all of th
 # if type(food.features["label"]) != 'datasets.features.features.ClassLabel': # Cast to ClassLabel
 #     food = food.class_encode_column('label')
 print(food)
-desired_label = 'x_hot_dog_box'
-desired_thickness = 1
+desired_label = 'x_plus_box'
+desired_thickness = 3
 desired_density = 1
 
 data_frame = pd.DataFrame(food)
@@ -60,7 +62,7 @@ all_shapes = [basic_box, diagonal_box_split, horizontal_vertical_box_split, back
                  x_hot_dog_box, x_plus_box]
 
 base_shapes = [basic_box, back_slash_box, forward_slash_box, hot_dog_box, hamburger_box]
-image_size = 11
+image_size = 256
 density = [1]
 
 boxes = make_boxes(image_size, density, all_shapes)