如何获取有关训练Tensorflow对象检测API的火车输入数据？

Question

当faster_rcnn_resnet101模型训练时，损耗显示在终端的每个步骤中。

我想知道每个步骤都输入了哪些数据。当损失增加时，我不知道为什么损失会增加。

有人知道如何在每个步骤中查看输入数据吗？

Answer 1

您无法检查每个步骤的结果，但是会在您的object_detection / training目录中创建经过训练的检查点前缀，该前缀将在没有特定步骤的情况下更新。 您可以使用当前训练有素的模型检查物体检测。

例如：

python3 export_inference_graph.py \
    --input_type image_tensor \
    --pipeline_config_path training/ssd_mobilenet_v1_pets.config \
    --trained_checkpoint_prefix training/model.ckpt-25000 \
    --output_directory latest_dataset

此处 model.ckpt-25000 是到目前为止所训练的步骤（25000）的编号。