我已经使用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
如何通过消除此错误将模型导出到冻结图形中?