Numpy判断某维度是否相同
在 语义分割(Semantic segmentation) 领域深度学习的label预处理工作中,要把3通道的label数据转为单通道的mask作为训练样本。例如把一堆黑色和红色的像素,转为全是0和1的单通道 mask。最Normal的处理流程为:判断像素的颜色;根据颜色赋值对应的label。那么在步骤一中,就要判断 label 的 channel 维度(例如shape为(480, 1280, 3),则channel维度为最后的3)和 color是否相同。假设 label 大小为 4 x
原创
2020-09-18 18:25:30 ·
1444 阅读 ·
0 评论