Hi, thanks for interest,

1. you can use '--image=path_image' parameter to run custom images.. (its important keep the same pattern between custom image and trained model)

2. the code use checkpoint callback already to save the model in hdf5 (best weigths) and load as well. I think you can use save and load functions from TF too, but I didn't try