0

我从这里使用 AlexNet

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

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

4

2 回答 2

0

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

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

于 2018-04-28T16:16:21.417 回答
0

你需要

在预训练的网络之上再添加一层

. 这将是你的

输出层

.

1000 个类层的这个输出将是这个层的输入，它会给你的 3 个类作为输出。

之后用你的图像训练这个新网络

于 2018-04-28T20:04:50.003 回答