如何获得验证损失以及张量板中的总损失

时间:2019-09-27 09:54:37

标签: tensorflow deep-learning object-detection tensorboard faster-rcnn

我正在使用预训练的模型来使用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,与预测损失的验证图像比较以及总损失,最后失利。 我阅读了一些有关如何包括herehere之类的验证丢失的资源,但是我不知道应将这些代码行附加到哪个文件,甚至不知道该在我的哪个文件中进行编辑夹。有什么建议么?

0 个答案:

没有答案