时间戳

首页 > 解决方案 > BoxClassifierLoss/localization_loss 和 Loss/regularization_loss 使用 Tensorflow Objectdetection API 也为零

deep-learning - BoxClassifierLoss/localization_loss 和 Loss/regularization_loss 使用 Tensorflow Objectdetection API 也为零

问题描述

我现在一直在尝试训练一个对象检测模型，即“faster_rcnn_resnet152_v1_640x640_coco17_tpu-8”模型。然而，在整个训练过程中，BoxClassifierLoss/localization_loss 和 Loss/regularization_loss 都没有高于零。

有没有其他人遇到过类似的问题，或者有人知道解决方案吗？（我认为这就是我的模型至少表现非常糟糕的原因）

INFO:tensorflow:{'Loss/BoxClassifierLoss/classification_loss': 0.011540242,
 'Loss/BoxClassifierLoss/localization_loss': 0.0,
 'Loss/RPNLoss/localization_loss': 0.05603733,
 'Loss/RPNLoss/objectness_loss': 0.021345321,
 'Loss/regularization_loss': 0.0,
 'Loss/total_loss': 0.08892289,
 'learning_rate': 0.090500005}
I1105 01:40:26.982768 16300 model_lib_v2.py:705] {'Loss/BoxClassifierLoss/classification_loss': 0.011540242,
 'Loss/BoxClassifierLoss/localization_loss': 0.0,
 'Loss/RPNLoss/localization_loss': 0.05603733,
 'Loss/RPNLoss/objectness_loss': 0.021345321,
 'Loss/regularization_loss': 0.0,
 'Loss/total_loss': 0.08892289,
 'learning_rate': 0.090500005}

标签： deep-learningobject-detectiontensorloss-functionobject-detection-api

解决方案

'localization_loss' 始终保持在 0.0 可能是由于 tfrecords 文件中的错误，或者很可能是 label_map 中的错误。检查您的 label_map 是否与 tfrecords 文件中的类匹配并且格式正确。

推荐阅读