np.all的实例应用

gt_image是一张RGB的label图，背景颜色是【0,0,0】，目标颜色是【255,0,0】

background_color=[255,0,0]
road_color=[0,0,0]
#这样处理会将比如说【2,4,3】变为【2,4】，将3-D变为2-D。
gt_bg=np.all(gt_image==background_color,axis=2)
gt_road=np.all(gt_image==road_color,axis=2)

assert(gt_road.shape==gt_bg.shape)

#有效区域为True,无效区域也就是干扰（噪音）为False
valid_gt=gt_road+gt_bg

shape=gt_bg.shape
# 将二维的数据变成三维
gt_bg=gt_bg.reshape(shape[0],shape[1],1)
gt_road=gt_road.reshape(shape[0],shape[1],1)
# 将这两个进行拼接，第一层是背景第二层是目标图并且都是bool值也就是二值图像
gt_image=np.concatenate((gt_bg,gt_road),axis=2)