首页 > 解决方案 > 在 tensorflow 对象检测 api 中训练 softmax 层

问题描述

我正在尝试使用tensorflow 对象检测 api训练库存跟踪应用程序,并且我使用了本教程。我的图像数据集太小,无法训练神经网络中的所有权重,我想训练后面的几层,甚至只训练 softmax 层。但是我没有找到任何教程告诉我如何声明要训练的层。我怎样才能做到这一点?

谁能给我一个关于这个的链接或github问题?

标签: pythontensorflowcomputer-visionobject-detectionobject-detection-api

解决方案


推荐阅读