Commit 22053865 authored by mihaivanea's avatar mihaivanea
Browse files

Don't have labeled data for FDDB to do object localisation, therefore I need...

Don't have labeled data for FDDB to do object localisation, therefore I need to change the model so that I do cross-validation only on the regression layer.
parent 41fcb5bf
#!/vol/bitbucket/mv1315/urop/venv/bin/python3.5
from keras.layers import Dense, Input
from keras.models import Model
from generate_arrays_fddb import load_arrays_fddb
inputs = Input(shape=(224, 224,3,))
x = Dense(100, activation="relu")(inputs)
x = Dense(100, activation="relu")(x)
predictions = Dense(4, activation="softmax")(x)
model = Model(inputs=inputs, outputs=predictions)
model.compile(
optimizer="rmsprop",
loss="mean_squared_error",
matrics=["accuracy"])
fddb_path_train = "../fddb/FDDB-folds/FDDB-fold-01-ellipseList.txt"
x_train, y_train = load_arrays_fddb(fddb_path_train)
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