MMlab中遥感数据集二分类标签无法读取问题的原因及解决

原因

遥感数据集的二分类标签通常是单通道数据,无论格式是tif,jpg,png,对于单通道数据,像素值要和类别的id相等,比如’background‘,’building‘在单通道的label数据中像素值分别要为0,1,但由于数据’background‘=0,’building‘=255,所以不能读取。

解决办法

在mmseg/datasets/pipelines/loading.py中143行修改源码:

gt_semantic_seg[gt_semantic_seg_copy == old_id*255] = new_id

你可能感兴趣的:(python,深度学习,pytorch)