!saved_model_cli show --dir {export_path} --all
Example:
saved_model_cli show --dir train_strawberrydc_iter5_labEq_combo_augmented_both_167/saved_model --all
To check versions
pip show keras
Name: Keras
Version: 2.1.2
pip show tensorflow-gpu
Name: tensorflow-gpu
Version: 1.3.0
Mask R-CNN
# How to convert COCO model (.h5) to tensorflow .pb mode
Ref: https://github.com/matterport/Mask_RCNN/issues/218
To produce h5:
python3 coco.py evaluate --dataset=$COCO_PATH --model=coco
To save model in coco.py:
evaluate_coco(model, dataset_val, coco, "bbox", limit=int(args.limit)) model.keras_model.save("mrcnn_eval.h5")
Extracting pb from h5:
python3 keras_to_tensorflow.py -input_model_file saved_model_mrcnn_eval.h5 -output_model_file model.pb -num_outputs=7