tensorflow - 配置中的类比训练的类多(tensorflow 对象检测 API)

标签 tensorflow object-detection-api

我从 tensorflow 的 OD API 继续训练了一个 fast_rcnn_inception 网络,该网络在我的仅包含一个类的数据集上提供了检查点。

我根据自己的需要更改了配置文件并进行了成功的培训。
测试我训练有素的网络给了我很好的准确性结果。

但是现在我意识到在配置文件中我忘记切换 num_classes从 90 到 1。

这对我的网络有什么影响?如果配置正确设置为 1,它现在是否比它慢?

tensorflow 的 API 是否根据提供的 num_classes 自动分配过滤器的数量/大小? ?

最佳答案

我会回答我自己的问题:

我用正确的 num_classes 进行了再培训设置为 1 并且只有很小的明显差异:

  • 检查点现在小 3MB(103 而不是 106MB)。
  • 对于 num_classes 错误地设置为 90,mAP 有一条更平滑的线

    num_classes set to 1
    num_classes set to 90

  • 但我仍然不知道它在 tensorflow 内部有什么不同,但这似乎并不重要。

    关于tensorflow - 配置中的类比训练的类多(tensorflow 对象检测 API),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48497407/

    相关文章:

    Tensorflow 对象检测 : visualization_export_dir in eval_config not saving eval images

    Tensorflow 对象检测 API 无效参数 : Shape mismatch in tuple component 16. 预期 [1,?,?,3],得到 [1,182,322,4]

    python - 如何用numpy实现tf.space_to_depth?

    python - 如何让 estimator.predict 预测一个样本

    tensorflow - 关于tensorflow lite量化代码、论文和文档不一致的问题

    machine-learning - 如何使卷积神经网络接受可变大小的输入?

    python - Keras:两个同时层,其中一个对前一层的输出进行卷积

    python - 类型错误 cv2.rect

    python - Tensorflow 对象检测 : how to detect on batch

    tensorflow - 使用对象检测 API 的训练未在 AI Platform 中的 GPU 上运行