tensorflow - 如何为 3 个类而不是 1000 个类自定义 AlexNet？

Question

我从这里使用 AlexNet

那里的数据有 1000 个类别，因此它具有相应的权重。如何让它用于预测具有 3 个类的数据的值？

我知道我必须改变重量，但我不知道怎么做。

标签： tensorflowneural-networkdeep-learningconv-neural-networkconvolutional-neural-network

Answer 1

您只需设置num_classes = 3which 将减少模型输出张量和单独定义的 placeholder 的输出类的数量y。

调用时会相应调整权重的数量，即参数model = AlexNet(...。