尝试从传输培训创建的自定义检查点创建tflite pb文件时遇到错误
系统信息:
ubuntu - 18.0
tf.VERSION = 1.14.1-dev20190517
tf.GIT_VERSION = v1.12.1-2154-g3df6d99f3f
tf.COMPILER_VERSION = v1.12.1-2154-g3df6d99f3f
env : LD_LIBRARY_PATH /usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/lib64
Tensorflow命令:
python /object_detection/export_tflite_ssd_graph.py
--pipeline_config_path models/ssd_mobilenet_v1_pets.config
--trained_checkpoint_prefix model.ckpt-23595
--output_directory /tflite/
--add_postprocessing_op=true
错误日志:
TypeError:
export_tflite_graph()
接受6个位置参数,但给出了7个
任何帮助表示赞赏!
答案 0 :(得分:0)
这是脚本的错误。我们会尽快修复。 感谢您举报!