marta-marta
commited on
Commit
•
a3e02cb
1
Parent(s):
d1454c6
Addressing type hint error
Browse files
Data_Generation/Dataset_Generation_Functions.py
CHANGED
@@ -9,14 +9,14 @@ import matplotlib.pyplot as plt
|
|
9 |
from json import JSONEncoder
|
10 |
|
11 |
|
12 |
-
|
13 |
########################################################################################################################
|
14 |
# Make the data using all the code in Shape_Generation_Functions.py
|
15 |
-
def make_boxes(image_size: int, densities: list
|
16 |
"""
|
17 |
:param image_size: [int] - the pixel height and width of the generated arrays
|
18 |
:param densities: [list[float]] - of the desired pixel values to apply to active pixels - Recommend values (0,1]
|
19 |
-
:return:
|
|
|
20 |
"""
|
21 |
|
22 |
matrix = []
|
|
|
9 |
from json import JSONEncoder
|
10 |
|
11 |
|
|
|
12 |
########################################################################################################################
|
13 |
# Make the data using all the code in Shape_Generation_Functions.py
|
14 |
+
def make_boxes(image_size: int, densities: list) -> list:
|
15 |
"""
|
16 |
:param image_size: [int] - the pixel height and width of the generated arrays
|
17 |
:param densities: [list[float]] - of the desired pixel values to apply to active pixels - Recommend values (0,1]
|
18 |
+
:return: list[tuple] - [Array, Density, Thickness of each strut type] this is all the defining information for
|
19 |
+
all the generated data.
|
20 |
"""
|
21 |
|
22 |
matrix = []
|