标签： tensorflow neural-network deep-learning conv-neural-network convolutional-neural-network
我正在使用来自here
那里的数据有1000个类，所以根据它有权重。如何使其用于预测具有3个类的数据的值？
我知道我必须改变重量，但我不知道如何。
答案 0 :(得分：0)
您只需设置num_classes = 3，这将减少模型输出张量和单独定义的占位符y的输出类数。
num_classes = 3
y
调用model = AlexNet(...时，权重数量即参数将相应调整。
model = AlexNet(...
答案 1 :(得分：0)
你需要
在预先训练好的网络上添加一层
。这将是你的
输出图层
1000类图层的输出将是此图层的输入，它将为您的3个类提供输出。
之后用图像训练这个新网络