无法将Deeplab模型导出到冻结图

时间:2019-04-16 09:43:45

标签: python tensorflow machine-learning computer-vision

我已经使用Deeplab和Mobilenet训练了自己的分割模型,以从图像中分割出正方形。我想将此模型导出到冻结图。但是,当我运行Original Application Version脚本时,出现错误:

exported_model.py

使用的培训和评估脚本:

File is too short to be an sstable 

一切都很好,花花公子,而且我能够使用上面的脚本来可视化我的结果。但是当我必须使用export_model.py创建一个冻结的推理图时,就会遇到如下麻烦:我运行此脚本

NUM_ITERATIONS=5000
CKPT_NAME="deeplabv3_mnv2_pascal_train_aug"
CKPT_PATH="${TRAIN_LOGDIR}/model.ckpt-${NUM_ITERATIONS}"
EXPORT_PATH="${EXPORT_DIR}/frozen_inference_graph.pb"

=== Train the network ===
python "${WORK_DIR}"/train.py
--logtostderr
--train_split="train"
--model_variant="mobilenet_v2"
--output_stride=16
--train_crop_size=513
--train_crop_size=513
--train_batch_size=4
--training_number_of_steps="${NUM_ITERATIONS}"
--dataset="testset"
--tf_initial_checkpoint="${INIT_FOLDER}/${CKPT_NAME}/model.ckpt-30000.index"
--train_logdir="${TRAIN_LOGDIR}"
--dataset_dir="${FINAL_DATASET}"
--initialize_last_layer=False
--last_layers_contain_logits_only=False
--fine_tune_batch_norm=False

=== Run evaluation ===
python "${WORK_DIR}"/eval.py
--logtostderr
--eval_split="val"
--model_variant="mobilenet_v2"
--eval_crop_size=513
--eval_crop_size=513
--dataset="testset"
--checkpoint_dir="${TRAIN_LOGDIR}"
--eval_logdir="${EVAL_LOGDIR}"
--dataset_dir="${FINAL_DATASET}"
--max_number_of_evaluations=1

=== Visualize the results ===
python "${WORK_DIR}"/vis.py
--logtostderr
--vis_split="val"
--model_variant="mobilenet_v2"
--vis_crop_size=513
--vis_crop_size=513
--dataset="testset"
--checkpoint_dir="${TRAIN_LOGDIR}"
--vis_logdir="${VIS_LOGDIR}"
--dataset_dir="${FINAL_DATASET}"
--max_number_of_iterations=1

我收到以下错误:

python "${WORK_DIR}"/export_model.py
--logtostderr
--checkpoint_path="${CKPT_PATH}"
--export_path="${EXPORT_PATH}"
--model_variant="mobilenet_v2"
--num_classes=3
--crop_size=513
--crop_size=513
--inference_scales=1.0

如何通过消除此错误将模型导出到冻结图形中?

0 个答案:

没有答案