自定义数据集上的tensorflow对象检测API评估

Question

我有一个使用tensorflow对象检测api [Faster RCNN]训练的自定义数据集。

我保存了模型检查点。现在，我想在我的评估集上运行该模型。但是，我找不到用于在评估集上运行的脚本。 我有efset的tfrecord文件。

如何从对象检测api中为自己保存的模型检查点运行eval脚本？我找不到脚本。

Answer 1

您正在寻找eval.py。

使用以下命令运行此脚本：

python eval.py \
--logtostderr \
--pipeline_config_path=PATH_TO_YOUR_CONFIG_FILE \
--checkpoint_dir=PATH_TO_CHECKPOINT \
--eval_dir=OUTPUT_PATH