AssertionError: Checkpoint not found.

I have tensorflow version 1.11.0

When I save the weights as a checkpoint using model.save_weights(checkpoints_path + "." + str(ep))

the file is saved as

path_to_checkpoints.0.data-00000-of-00001

not as

path_to_checkpoints.0

and when I apply

python -m keras_segmentation predict \
 --checkpoints_path="path_to_checkpoints" \
 --input_path="dataset1/images_prepped_test/" \
 --output_path="path_to_predictions"

I got the following error.

AssertionError: Checkpoint not found.

[Shiny-ZhangXiXin]

I solved this problem, you can load the model without using the function written by the author, maybe there are some bugs in it, right?The solution is as follows: in keras_segmentation/models/predict.py Comment out the following code:

Assert (os.path.isfile(checkpoints_path+"_config.json"), "Checkpoint not found."
Model_config = json. Loads (open (checkpoints_path + "_config. Json", "r"), read ())

To: Model_config = json. Loads (open (". / checkpoints/vgg_unet_1_config. Json ", "r"), read ()) Inside the quotes is the path to your json file, either absolute or relative The same thing with this one: Latest_weights = find_latest_checkpoint (checkpoints_path) To: Latest_weights = "checkpoints/vgg_unet_1. 22" Use your model file in quotes and use whatever model you want

[lyc1995452-star]

I solved it using the following command

model.save_weights(checkpoints_path + "." + str(ep), save_format='h5')

Where should I put this?Whether to retrain？