tensorflow - TensorFlow 对象检测 API Num 类

Question

我想知道 Tensorflow 对象检测 API 中 num_classes 的值是否会影响模型性能。例如，假设我正在训练模型以仅检测一个对象（例如人类），但在 num_classes 中我将其设置为 20。不正确的 num_classes 数量会影响模型性能吗？

标签： tensorflowobject-detection-api

Answer 1

如果您设置num_classes为高于实际的类数，这并不重要（它只会使一些张量/变量更大一点），但如果您将其设置为较低，则很重要。第一种情况不应影响模型性能。