Commit 135ec5d2 authored by mihaivanea's avatar mihaivanea
Browse files

Fixed the dimensionality problem with the labelled data

parent 461c0cc0
......@@ -15,12 +15,10 @@ model = Model(inputs=inputs, outputs=predictions)
model.compile(
optimizer="rmsprop",
loss="mean_squared_error",
matrics=["accuracy"])
metrics=["accuracy"])
fddb_path_train = "../fddb/FDDB-folds/FDDB-fold-01-ellipseList.txt"
x_train, y_train = load_arrays_fddb(fddb_path_train)
print(x_train.shape)
print(y_train.shape)
model.fit(x_train, y_train)
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