Made edits to match the changes made in Piecewise_Box_Functions

Data_Generation/Shape_Generation_Functions.py

@@ -8,7 +8,7 @@ from Data_Generation.Piecewise_Box_Functions import back_slash_array, basic_box_
 def basic_box(additional_pixels, density, image_size):
     A = basic_box_array(image_size)  # Creates the outside edges of the box
     # Increase the thickness of each part of the box
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A*density
 
 
@@ -18,7 +18,7 @@ def horizontal_vertical_box_split(additional_pixels, density, image_size):
     A = update_array(A, hot_dog_array(image_size), image_size)
     A = update_array(A, hamburger_array(image_size), image_size)
     # Increase the thickness of each part of the box
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A*density
 
 
@@ -31,35 +31,35 @@ def diagonal_box_split(additional_pixels, density, image_size):
 
     # Adds pixels to the thickness of each component of the box
     # Increase the thickness of each part of the box
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A*density
 
 
 def back_slash_box(additional_pixels, density, image_size):
     A = basic_box_array(image_size)  # Initializes A matrix with 0 values
     A = update_array(A, back_slash_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
 def forward_slash_box(additional_pixels, density, image_size):
     A = basic_box_array(image_size)  # Initializes A matrix with 0 values
     A = update_array(A, forward_slash_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
 def hot_dog_box(additional_pixels, density, image_size):
     A = basic_box_array(image_size)  # Initializes A matrix with 0 values
     A = update_array(A, hot_dog_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
 def hamburger_box(additional_pixels, density, image_size):
     A = basic_box_array(image_size)  # Initializes A matrix with 0 values
     A = update_array(A, hamburger_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
@@ -69,7 +69,7 @@ def x_plus_box(additional_pixels, density, image_size):
     A = update_array(A, hamburger_array(image_size), image_size)
     A = update_array(A, forward_slash_array(image_size), image_size)
     A = update_array(A, back_slash_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
@@ -79,7 +79,7 @@ def forward_slash_plus_box(additional_pixels, density, image_size):
     A = update_array(A, hamburger_array(image_size), image_size)
     A = update_array(A, forward_slash_array(image_size), image_size)
     # A = update_array(A, back_slash_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
@@ -88,7 +88,7 @@ def back_slash_plus_box(additional_pixels, density, image_size):
     A = update_array(A, hot_dog_array(image_size), image_size)
     A = update_array(A, hamburger_array(image_size), image_size)
     A = update_array(A, back_slash_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
@@ -97,7 +97,7 @@ def x_hot_dog_box(additional_pixels, density, image_size):
     A = update_array(A, hot_dog_array(image_size), image_size)
     A = update_array(A, forward_slash_array(image_size), image_size)
     A = update_array(A, back_slash_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
@@ -106,7 +106,7 @@ def x_hamburger_box(additional_pixels, density, image_size):
     A = update_array(A, hamburger_array(image_size), image_size)
     A = update_array(A, forward_slash_array(image_size), image_size)
     A = update_array(A, back_slash_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
@@ -115,7 +115,7 @@ def forward_slash_hot_dog_box(additional_pixels, density, image_size):
     A = basic_box_array(image_size)  # Initializes A matrix with 0 values
     A = update_array(A, hot_dog_array(image_size), image_size)
     A = update_array(A, forward_slash_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
@@ -123,7 +123,7 @@ def forward_slash_hamburger_box(additional_pixels, density, image_size):
     A = basic_box_array(image_size)  # Initializes A matrix with 0 values
     A = update_array(A, hamburger_array(image_size), image_size)
     A = update_array(A, forward_slash_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
@@ -131,7 +131,7 @@ def back_slash_hamburger_box(additional_pixels, density, image_size):
     A = basic_box_array(image_size)  # Initializes A matrix with 0 values
     A = update_array(A, hamburger_array(image_size), image_size)
     A = update_array(A, back_slash_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density
 
 
@@ -139,5 +139,5 @@ def back_slash_hot_dog_box(additional_pixels, density, image_size):
     A = basic_box_array(image_size)  # Initializes A matrix with 0 values
     A = update_array(A, hot_dog_array(image_size), image_size)
     A = update_array(A, back_slash_array(image_size), image_size)
-    A = add_pixels(A, additional_pixels, image_size)
+    A = add_pixels(A, additional_pixels)
     return A * density