在Pyplot中显示预测分割输出数据

import matplotlib.pyplot as plt
import numpy as np

output_data = np.load('output_data.npy')  # 假设数据存储为.npy格式的文件

plt.figure(figsize=(8, 8))  # 创建一个8x8英寸大小的图形窗口

# 绘制预测分割输出
plt.imshow(output_data, cmap='gray')  # 使用灰度颜色映射

# 添加标题和标签
plt.title('Predicted Segmentation Output')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')

# 显示颜色条
plt.colorbar()

# 显示图形
plt.show()