Commit 0d4d9b61 authored by mihaivanea's avatar mihaivanea
Browse files

Redid the numpy arrays

parent c1895ed8
......@@ -28,6 +28,7 @@ def convert_to_cartesian(ellip_array):
y2 = c_y + major_r
box = [round(p, 4) for p in (x1, y1, x2, y2)]
cartesian_output.append(box)
cartesian_output = np.array(cartesian_output)
return cartesian_output
def generate_arrays_fddb(path):
......@@ -55,12 +56,11 @@ def load_arrays_fddb(path):
data = np.asarray(img, dtype="float32")
if len(data.shape) == 3:
x.append(data)
y.append(convert_to_cartesian(lb[2]))
result = convert_to_cartesian(lb[2])
y.append(result)
x = np.array(x)
print(x.shape)
y = np.array(y)
print(y.shape)
return x
return x, y
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment