marta-marta commited on
Commit
577e72e
·
1 Parent(s): 822d985

Added new functions for additional shapes that were missed

Browse files
Data_Generation/Shape_Generation_Functions.py CHANGED
@@ -109,3 +109,35 @@ def x_hamburger_box(additional_pixels, density, image_size):
109
  A = add_pixels(A, additional_pixels, image_size)
110
  return A * density
111
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
109
  A = add_pixels(A, additional_pixels, image_size)
110
  return A * density
111
 
112
+
113
+ # New Unit Cells
114
+ def forward_slash_hot_dog_box(additional_pixels, density, image_size):
115
+ A = basic_box_array(image_size) # Initializes A matrix with 0 values
116
+ A = update_array(A, hot_dog_array(image_size), image_size)
117
+ A = update_array(A, forward_slash_array(image_size), image_size)
118
+ A = add_pixels(A, additional_pixels, image_size)
119
+ return A * density
120
+
121
+
122
+ def forward_slash_hamburger_box(additional_pixels, density, image_size):
123
+ A = basic_box_array(image_size) # Initializes A matrix with 0 values
124
+ A = update_array(A, hamburger_array(image_size), image_size)
125
+ A = update_array(A, forward_slash_array(image_size), image_size)
126
+ A = add_pixels(A, additional_pixels, image_size)
127
+ return A * density
128
+
129
+
130
+ def back_slash_hamburger_box(additional_pixels, density, image_size):
131
+ A = basic_box_array(image_size) # Initializes A matrix with 0 values
132
+ A = update_array(A, hamburger_array(image_size), image_size)
133
+ A = update_array(A, back_slash_array(image_size), image_size)
134
+ A = add_pixels(A, additional_pixels, image_size)
135
+ return A * density
136
+
137
+
138
+ def back_slash_hot_dog_box(additional_pixels, density, image_size):
139
+ A = basic_box_array(image_size) # Initializes A matrix with 0 values
140
+ A = update_array(A, hot_dog_array(image_size), image_size)
141
+ A = update_array(A, back_slash_array(image_size), image_size)
142
+ A = add_pixels(A, additional_pixels, image_size)
143
+ return A * density