Fix RasterLabelVisualizer to not use antialiasing that led to artifacts around masks.

This commit is contained in:
Siyu Yang 2020-11-04 14:20:32 -08:00
Родитель 3923cf0f70
Коммит 365ca160b4
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -258,7 +258,7 @@ class RasterLabelVisualizer(object):
assert np.max(label_raster) <= self.num_classes, f'Invalid value for class label: {np.max(label_raster)}'
_ = plt.figure(figsize=size)
_ = plt.imshow(label_raster, cmap=self.colormap, norm=self.normalizer)
_ = plt.imshow(label_raster, cmap=self.colormap, norm=self.normalizer, interpolation='none')
buf = BytesIO()
plt.savefig(buf, format='png')