SSDmobilenet宠物配置->尝试从传输培训创建的自定义检查点创建tflite PB文件时遇到错误

时间:2019-05-20 08:22:16

标签: tensorflow

尝试从传输培训创建的自定义检查点创建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个

任何帮助表示赞赏!

1 个答案:

答案 0 :(得分:0)

这是脚本的错误。我们会尽快修复。 感谢您举报!