TensorFlow Object Detection API - what do the losses mean in the object detection api?

让人想犯罪 __ 提交于 2019-11-27 17:55:49

问题


What do each for the following losses mean? (in the TensorFlow Object detection API, while training FasterRCNN based models)

Loss/BoxClassifierLoss/classification_loss/mul_1

Loss/BoxClassifierLoss/localization_loss/mul_1

Loss/RPNLoss/localization_loss/mul_1

Loss/RPNLoss/objectness_loss/mul_1

clone_loss_1


回答1:


The losses for the Region Proposal Network:

Loss/RPNLoss/localization_loss/mul_1: Localization Loss or the Loss of the Bounding Box regressor for the RPN

Loss/RPNLoss/objectness_loss/mul_1: Loss of the Classifier that classifies if a bounding box is an object of interest or background

The losses for the Final Classifier:

Loss/BoxClassifierLoss/classification_loss/mul_1: Loss for the classification of detected objects into various classes: Cat, Dog, Airplane etc

Loss/BoxClassifierLoss/localization_loss/mul_1: Localization Loss or the Loss of the Bounding Box regressor




回答2:


clone_loss_1 is relevant only if you train on multiple GPUs: Tensorflow would create a clone of the model to train on each GPU and report the loss on each clone. If you are training the model on a single GPU/CPU, then you will just see clone_loss_1, which is the same as TotalLoss.

The other losses are as described in Rohit's answer.



来源:https://stackoverflow.com/questions/48111847/tensorflow-object-detection-api-what-do-the-losses-mean-in-the-object-detectio

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!