Commit 9ea6a518 authored by mihaivanea's avatar mihaivanea
Browse files

Generator works. Need to process the images to feed them as training input to fit().

parent 383a8654
......@@ -31,16 +31,13 @@ model.compile(optimizer='rmsprop', loss='categorical_crossentropy')
fddb_path = "../fddb/FDDB-folds/FDDB-fold-01-ellipseList.txt"
# train the model on the new data for a few epochs
fail = 0
try:
model.fit_generator(
generator=generate_arrays_fddb(fddb_path),
steps_per_epoch=1000, epochs=10
)
except Exception as e:
fail += 1
print(e)
print("FAIL:" + str(fail))
# at this point, the top layers are well trained and we can start fine-tuning
# convolutional layers from inception V3. We will freeze the bottom N layers
......
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