Python中文
首页
教程
问答
标签
搜索
登录
注册
使用时出现InvalidArgumentErrortensorflow.data.Dataset.从安根
回答此问题可获得
20
贡献值,回答如果被采纳可获得
50
分。
<p>我尝试在单个GPU上使用Tensorflow数据集API生成自己的图像数据集来测量推理性能</p> <pre><code>resolutions = [ (2048, 1080) ] def generate_image(size, channels): image_value = random.random() image_shape = [1, size[1], size[0], channels] return tf.constant( value=image_value, shape=image_shape, dtype=tf.float32) def generate_single_input(size): source = generate_image(size, 3) target = generate_image(size, 3) return source, target def input_generator_fn(): for res in resolutions: for i in range(10): yield generate_single_input(res) def benchmark(): ... ds = tf.data.<a href="https://www.cnpython.com/pypi/dataset" class="inner-link">dataset</a>.from_generator( generator=input_generator_fn, output_types=(tf.float32, tf.float32), output_shapes=(tf.TensorShape([1, 1080, 2048, 3]), tf.TensorShape([1, 1080, 2048, 3]))) iterator = ds.make_one_shot_iterator() next_record = iterator.get_next() inputs = next_record[0] outputs = next_record[1] predictions = { 'input_images': inputs 'output_images': outputs } session = tf.Session() with session: tf.global_variables_initializer() for res in resolutions: for i in range(10): session.run(predictions) ..... </code></pre> <p>但我在运行后<strong>观察到以下异常</strong>:</p> ^{pr2}$
0 条评论
分类:
Python问答
请先
登录
后评论
默认排序
时间排序
1 个回答
匿名
1天前
擅长:python、mysql、java
<p>你知道了吗?在</p> <p>我遇到了完全相同类型的问题,我的问题是生成器和输入到输出形状中的尺寸不匹配。在</p> <p>再看看你的代码,我相信你必须输入有效的数据,比如numpy数组,而不是TensorFlow常量。在</p>
请先
登录
后评论
针对此问题:
更多的回答
关注
89
关注
收藏
1
收藏,
216
浏览
网友 提问于 2天前
相关Python问题
如何在python中从数据帧列中删除分类值?
6 回答
如何在python中从数据帧列表中删除引号
6 回答
如何在python中从数据帧创建列表
10 回答
如何在Python中从数据帧创建嵌套的JSON
10 回答
如何在Python中从数据帧显示wordcloud
8 回答
如何在Python中从数据帧的时间戳中删除字符
10 回答
如何在Python中从数据帧绘制简单绘图?
5 回答
如何在python中从数据帧行提取具有特定长度的范围?
2 回答
如何在python中从数据帧设置dict中的值
1 回答
如何在Python中从数据库中获得一个结果
4 回答
如何在python中从数据框中绘制分类条形图
5 回答
如何在Python中从数据框中选择特定细节?
8 回答
如何在python中从数据集中删除unicode
7 回答
如何在python中从数据集中删除某些数值?
8 回答
如何在python中从数据集中选择行
5 回答
如何在Python中从数组中删除元素
3 回答
如何在python中从数组中删除单个倒逗号?
7 回答
如何在python中从数组中删除对象?
3 回答
如何在python中从数组中删除引号
7 回答
如何在python中从数组中删除所有最小值
4 回答