我使用以下代码调整语义分段任务[PASCAL VOC]的标签图像的大小(图片的值范围为0到n_classes
,n_classes
= 21)。
lbl = m.imread(lbl_path)
lbl = np.array(lbl, dtype=np.int32)
lbl = self.encode_segmap(lbl)
print 'Unique: ', np.unique(lbl), lbl.dtype
plt.imshow(lbl)
plt.show()
lbl = m.imresize(lbl, (self.img_size, self.img_size), 'nearest')
print 'Unique after resize: ', np.unique(lbl), lbl.dtype
plt.imshow(lbl)
plt.show()
当我运行它时,我得到了这个输出:
Unique: [ 0 1 15] int64
Unique after resize: [ 0 17 255] uint8
调整大小(使用nearest
)后的值是否只属于集合(0,1,15)?
我得到的图像是:
原
调整大小