model_h5_to_tflite3.py

params = {}
params.update({
    'img_size': 300,
    "n_all": 300,
    "epochs": 1,
    "batch_size": 64
})
params["df_labels_for_model"] = pd.read_csv("../csv/labels_for_model.csv")
params["len_labels"] = len(params["df_labels_for_model"])
params["n_photos"] = fetch_n_photos(params)

df_photo_ids = {}
for w in ["train", "test"]:
  filedir = "../csv/model_results/" + model_datetime + "/data_" + w + ".csv"
  df = pd.read_csv(filedir)
  df = df.drop(df.columns[[0]], axis=1)
  df_photo_ids[w] = df
params["len_labels"] = len(params["df_labels_for_model"])

list_df = add_batch_number_to_df(df_photo_ids['train'], params)
gen = generate_fit_data(list_df, params)

X_test, y_test = fetch_fit_data(df_photo_ids['test'].photo_id, df_photo_ids['test'].y, params["img_size"], params["df_labels_for_model"])