使用tf.reduce_mean计算图像

Question

当我尝试使用tf.reduce_mean计算所有轴的图像平均值时，它显示＆＃34;启动内核时发生错误＆＃34;。

这是我的代码：

import tensorflow as tf
import numpy as np
imgdata = tf.read_file("./test_img/00000001.jpg")
my_img = tf.image.decode_jpeg(imgdata)
image = tf.reduce_mean(my_img)
tf.Session().run(image)

我尝试运行会话以获取＆＃34; my_img＆＃34;在计算平均值之前，它也不起作用。

如果我尝试计算由我自己创建的具有相同形状的数组的平均值，那就顺利了。

my_array = = np.random.randn(720, 1280, 3)
tf.Session().run(tf.reduce_mean(my_array))

Answer 1

通过将数据类型从uint8转换为float32：

解决了这个问题

my_img = 255.0 * tf.image.convert_image_dtype(my_img, tf.float32)