“ValueError : Tensor Tensor("predictions/Softmax:0", shape=(?, 2), dtype=float32) is not an element

最新推荐文章于 2022-01-07 15:27:39 发布

ystsaan

最新推荐文章于 2022-01-07 15:27:39 发布

阅读量2.8k

点赞数 4

分类专栏：深度学习

本文链接：https://blog.csdn.net/weixin_42388228/article/details/89914795

版权

在使用Keras和TensorFlow框架训练神经网络后，尝试在C#多线程环境中调用Python代码进行预测时遇到错误：`ValueError: Tensor Tensor("predictions/Softmax:0", shape=(?, 2), dtype=float32) is not an element`。解决方案是加载模型后，设置默认图形上下文`tf.get_default_graph()`，并在预测前使用`with graph.as_default():`。通过这些修改，可以成功在多线程子线程中运行预测。" 121611224,2562453,Ghidra逆向分析：二进制程序解析与实战,"['网络安全', '渗透测试', '逆向分析', '软件安全', 'Ghidra工具']

摘要由CSDN通过智能技术生成

原始问题及解决方案
https://github.com/keras-team/keras/issues/2397#issuecomment-254919212

问题描述：
在keras+tensorflow框架下训练神经网络并得到权重h5文件。
在之后需要调用的python代码中读取权重和图片并预测
在C#多线程的子线程调用python代码时出现以下报错
“ValueError : Tensor Tensor(“predictions/Softmax:0”, shape=(?, 2), dtype=float32) is not an element

解决方法：
主要是在读取权重后增加一行self.graph = tf.get_default_graph()
self.model=tf.keras.models.load_model(‘C:\Users\18367\Desktop\corner_test\0324_lug_irv2.h5’)
self.graph = tf.get_default_graph()
并在需要预测时前加 with self.graph.as_default():

原始py文件代码

# coding: utf-8

# In[1]:


import numpy as np
import tensorflow as tf
import cv2
import os

#model = tf.keras.models.load_model('C:/Users/18367/Desktop/corner_test/test_0430.h5')
class MODEL_lug(object):
    def __init__(self):
        self.model=tf.