PASCAL-VOC2012数据集及其增强版本的处理问题

本篇主要总结语义分割处理PASCAL-VOC2012数据集以及遇到的问题（axis 2 is out of bounds for array of dimension 0）

处理数据集主要参考该博客
https://blog.csdn.net/pangyunsheng/article/details/87360238
在将原始pascal voc 2012数据集中label的三通道RGB图像，转化为8-bit的灰度png图像时使用官方代码时

`
def convert_from_color_segmentation(arr_3d):
  arr_2d = np.zeros((arr_3d.shape[0], arr_3d.shape[1]), dtype=np.uint8)
  palette = pascal_palette()

  for c, i in palette.items():
    m = np.all(arr_3d == np.array(c).reshape(1, 1, 3), axis=2)
    arr_2d[m] = i

  return arr_2d`

出现‘axis 2 is out of bounds for array of dimension 0’报错
仔细研究后发现应该是numpy版本不对，尝试python3.7+numpy1.14/1.15以及python3.7+numpy1.15版本后无果，感兴趣的朋友可以继续降低版本尝试
直接将for循环的代码改为为

    # print(arr_3d.shape)
    a = np.array(c).reshape(1, 1, 3)
    a = np.repeat(a, arr_3d.shape[0], axis=0)
    a = np.repeat(a, arr_3d.shape[1], axis=1)
    # print(arr_3d[:,:,:3].shape)
    m = np.all(arr_3d[:,:,:3]==a, axis=2)
    arr_2d[m] = i

运行成功 灰度图生成正确
希望对以后使用该数据集的朋友有所帮助
数据集下载是在一个网盘下载的懒得找连接了有需要的可以评论- -