在哪里可以找到模型使用的损失度量类型?

时间:2019-02-18 03:59:48

标签: tensorflow object-detection-api

enter image description here

我试图找出ssdlite_mobilenet_v2_coco中使用了哪些损失函数,但是我似乎找不到在回购或在线中指定的损失函数。这是我在训练期间的损失的Tensorboard,但未指定损失指标。

回购链接:https://github.com/tensorflow/models/tree/master/research/object_detection

我正在关注本教程: https://github.com/EdjeElectronics/TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10#8-use-your-newly-trained-object-detection-classifier

我尝试浏览配置文件和代码的某些部分,但没有看到它使用哪种类型的指标。

在仓库中可以找到用于每种类型的模型体系结构的损耗度量的位置吗?谢谢!

1 个答案:

答案 0 :(得分:0)

您可以看到config file中使用的损失函数。损失包括两个损失:用于分类的weighted_sigmoid和用于本地化的weighted_smooth_l1,每一个权重均为1。要查看这些损失的代码,可以检查loss functions here。请注意,总损失还有另一个要进行正则化的部分。