我正在使用预训练的模型来使用TensorFlow对象检测API训练自定义数据集。为此,我选择了Faster_RCNN_inception_v2_pets模型,并使用model_main.py进行了训练,以便同时进行训练和评估。
python model_main.py --pipeline_config_path=training/faster_rcnn_inception_v2_pets.config --model_dir=training/ --sample_1_of_n_eval_examples=1 --alsologtostderr
我正在使用PAScal_VOC指标进行评估。 但是,我对训练损失和验证损失都感兴趣,并且能够比较它们,但是tensorborad给我的只是训练损失,边界框IOU AP,与预测损失的验证图像比较以及总损失,最后失利。 我阅读了一些有关如何包括here,here之类的验证丢失的资源,但是我不知道应将这些代码行附加到哪个文件,甚至不知道该在我的哪个文件中进行编辑夹。有什么建议么?