灰度图像与RGB图像的互相转换

最新推荐文章于 2025-10-31 10:18:47 发布

原创

最新推荐文章于 2025-10-31 10:18:47 发布 · 4.5k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#Pytorch #语义分割

本文介绍了如何使用Pytorch进行灰度图像与RGB图像之间的转换，主要涉及编程过程中的关键代码细节。

记代码中的一些小细节：

解码代码：

def decode_segmap(label_mask, dataset, plot=False):
    """Decode segmentation class labels into a color image
    Args:
        label_mask (np.ndarray): an (M,N) array of integer values denoting
          the class label at each spatial location.
        plot (bool, optional): whether to show the resulting color image
          in a figure.
    Returns:
        (np.ndarray, optional): the resulting decoded color image.
    """
   
    n_classes = 6
    label_colours = get_mydata_labels()
    
 
    r = label_mask.copy()  # h x w
    g = label_mask.copy()  # h x w
    b = label_mask.copy()  # h x w
    for ll in range(0, n_cl