本文共 505 字，大约阅读时间需要 1 分钟。

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,无效区域也就是干扰（噪音）为Falsevalid_gt=gt_road+gt_bgshape=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)

转载地址：https://blog.csdn.net/zz2230633069/article/details/82432047 如侵犯您的版权，请留言回复原文章的地址，我们会给您删除此文章，给您带来不便请您谅解！