Commit 74e981e2 authored by mihaivanea's avatar mihaivanea
Browse files

Having trouble with the format of loaded cross validation arrays.

parent 72400bf9
......@@ -60,8 +60,6 @@ def load_arrays_fddb(path):
y.append([result])
x = np.array(x)
y = np.array(y)
print(x.shape)
print(y.shape)
return x, y
#generator = generate_arrays_fddb(path)
......@@ -74,4 +72,4 @@ def load_arrays_fddb(path):
#for g in generator:
# print(g)
#print("DONE")
load_arrays_fddb(path)
#load_arrays_fddb(path)
......@@ -48,7 +48,7 @@ model = Model(
# Freeze the convolutional layers of ResNet
# Train only the two added top layers.
for layer in base_model.layer:
for layer in base_model.layers:
layer.trainable = False
# compile the modl
......@@ -69,7 +69,7 @@ sgd = SGD(lr=0.001, momentum=0.0, decay=0.0, nesterov=False)
#classifier_loss = mean_squared_error()
model.compile(
optimizer=[rms, sgd],
optimizer=rms,
loss=["categorical_crossentropy", "mean_squared_error"])
# Load the new model on FDDB.
......
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