keras相同模型,相同输入,不同机器输出完全不同的原因!!

问题关键:keras版本导致,不同机器keras版本不一致,将不同机器的keras版本保持一致就行了...

 最近在做一个以图搜图的项目,想到可以用keras的预训练模型提取特征,然后再计算各个图片特征的余弦距离。自己有差不多10万张图片,一开始在GPU服务器上提特征,然后将特征写在数据库里,在服务器上模拟识别效果还不错。然后自己在本地的电脑上模拟识别,先利用预训练模型提取特征然后再与GPU提取过的图片特征库进行比对,发现效果极差!

找了两天bug,发现了一个很难以置信的现象,同一张图片同一模型不同机器输出的结果相差极大!

然后发现了https://github.com/keras-team/keras/issues/7676这个网页,发现不止我一个人有这个现象..后来发现了GPU中keras版本为2.0.8,本地为2.20,将本地keras版本改为2.0.8,问题解决了..

果然版本控制真的是有道理的...

  • 3
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
TextCNN 模型是一种用于文本分类任务的深度学习模型,它基于卷积神经网络的思想,能够有效地提取文本特征并进行分类。本文将详细介绍 TextCNN 模型的原理和实现,以及如何使用 Keras 来搭建和训练 TextCNN 模型。 ## TextCNN 模型的原理 TextCNN 模型主要由以下三个部分组成: 1. **词嵌入层(Embedding Layer)**:将文本中的每个词转换为一个向量表示,以便在后续的卷积层中进行处理。这里我们可以使用预训练的词向量模型(如 GloVe、Word2Vec 等)来初始化词嵌入矩阵,也可以随机初始化一个词嵌入矩阵并在训练过程中进行优化。 2. **卷积层(Convolutional Layer)**:对词嵌入矩阵进行卷积操作,以提取不同大小的文本特征。这里我们可以使用多个不同大小的卷积核来提取不同尺寸的特征,然后将它们拼接起来得到一个更加丰富的特征表示。 3. **池化层(Pooling Layer)**:对卷积层输出的特征矩阵进行池化操作,以减少特征维度并提高模型的泛化能力。这里我们可以使用最大池化或平均池化等不同的池化方式。 最后,将池化层的输出连接到一个或多个全连接层中,以进行文本分类或其他相关任务。 ## TextCNN 模型Keras 实现 在 Keras 中,我们可以通过定义一个 `Sequential` 模型,并添加不同的层来实现 TextCNN 模型。下面是一个简单的 TextCNN 模型的实现示例: ```python from keras.models import Sequential from keras.layers import Embedding, Conv1D, GlobalMaxPooling1D, Dense model = Sequential() # 添加词嵌入层 model.add(Embedding(input_dim=vocab_size, output_dim=embedding_dim, input_length=maxlen)) # 添加卷积层和池化层 model.add(Conv1D(filters=128, kernel_size=3, activation='relu')) model.add(GlobalMaxPooling1D()) # 添加全连接层和输出层 model.add(Dense(128, activation='relu')) model.add(Dense(num_classes, activation='softmax')) # 编译模型 model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy']) ``` 在上面的代码中,我们首先定义了一个 `Sequential` 模型,并添加了三个不同的层:词嵌入层、卷积层和池化层。其中,词嵌入层的输入维度为词汇表大小(`vocab_size`);词嵌入矩阵的维度为 `embedding_dim`;输入序列的长度为 `maxlen`。卷积层和池化层的参数可以调整,以满足不同的文本特征提取需求。 接着,我们添加了一个全连接层和一个输出层,其中全连接层的输出维度为 `128`,激活函数为 ReLU;输出层的维度为分类数(`num_classes`),激活函数为 softmax。 最后,我们使用 `compile` 方法编译模型,并指定优化器、损失函数和评估指标。 ## 总结 TextCNN 模型是一种简单而有效的用于文本分类任务的深度学习模型,其主要思想是利用卷积神经网络来提取文本特征并进行分类。在 Keras 中,我们可以很方便地搭建和训练 TextCNN 模型,以满足不同的文本分类需求。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值