P2.14 加载并显示绿色通道的图片 def load_green_image(basepath, image_id): image = np.zeros(shape=(512,512)) image[:,:] = imageio.imread(basepath + image_id + "_green" + ".png") return image fig = plt.figure() image = load_green_image('./dataset/train/', x_train[0][0]) plt.imshow(image, cmap='gray') print(image.shape) x = image[:,:,np.newaxis] print(x.shape)如何让解决
时间: 2023-11-27 21:49:55 浏览: 133
加载图片显示
4星 · 用户满意度95%
这段代码的问题在于:使用的是灰度图像的颜色映射(`cmap='gray'`),但是加载的是绿色通道的图像。因此,应该使用颜色映射 `cmap='Greens'` 来显示绿色通道的图像。
以下是修改后的代码:
```python
import matplotlib.pyplot as plt
import imageio
def load_green_image(basepath, image_id):
image = np.zeros(shape=(512, 512))
image[:, :] = imageio.imread(basepath + image_id + "_green" + ".png")
return image
fig = plt.figure()
image = load_green_image('./dataset/train/', x_train[0][0])
plt.imshow(image, cmap='Greens') # 注意这里使用的是 'Greens' 颜色映射
print(image.shape)
x = image[:, :, np.newaxis]
print(x.shape)
```
运行后,应该可以正确地显示出绿色通道的图像。
阅读全文