处理

Question

我随机调整图像的亮度，饱和度等。但经过调整后，img_data的值超过0-1，因此imshow将无效。

tf.InteractiveSession()
image_raw_data=tf.read_file('C:/Users/User/PycharmProjects/Neural_Network\\cat.jpg')
sess=tf.Session()
img_data=tf.image.decode_jpeg(image_raw_data,channels=3)
img_data=tf.image.convert_image_dtype(img_data,dtype=tf.float32)
img_data=tf.image.resize_images(img_data,[300,300],method=0)
img_data=tf.image.random_brightness(img_data,max_delta=32/255)
plt.imshow(img_data.eval())

错误是：

ValueError: Floating point image RGB values must be in the 0..1 range.

我是否应该知道如何正确转换图像以便显示？

Answer 1

我要做的仅仅是想象图像是在图像中添加最小值，然后除以最大值，以下列方式将图像夹在0和1之间：

import numpy as np
img_data_np = img_data.eval()
min_val = np.min(img_data_np)
max_val = np.max(img_data_np)
img_data_clamped = (img_data_np - min_val) / (max_val - min_val)
plt.imshow(img_data_clamped)