导出TensorFlow训练模型时遇--output_directory参数错误求助

问题分析 #

执行导出命令时出现FATAL Flags parsing error: flag --output_directory=None错误，核心原因是命令中使用了Windows系统的路径分隔符\，在Linux环境下\会被当作转义字符，导致路径参数被解析为空值。CUDA库的警告属于无GPU环境下的正常提示，可忽略。

修正后的命令 #

当前路径为~/Desktop/TensorFlow/workspace/training_demo，使用以下命令（替换所有\为Linux兼容的/，并确保脚本路径正确）：

sudo python3 ../models/research/object_detection/exporter_main_v2.py \
--input_type image_tensor \
--pipeline_config_path models/my_ssd_resnet50_v1_fpn/pipeline.config \
--trained_checkpoint_dir models/my_ssd_resnet50_v1_fpn/ \
--output_directory exported-models/my_model

关键调整说明 #

• 路径分隔符替换：将所有\改为/，避免转义导致参数失效。
• 脚本路径验证：如果exporter_main_v2.py不在当前目录的上级路径，可替换为绝对路径，例如~/Desktop/TensorFlow/models/research/object_detection/exporter_main_v2.py。
• 目录存在性：确保models/my_ssd_resnet50_v1_fpn/目录下存在训练好的检查点文件（如ckpt-*），exported-models/my_model目录会由命令自动创建。

内容的提问来源于stack exchange，提问作者nobody