Сохранение значений массива numpy в пикселях изображения jpg

Question

Я работаю над проблемой сегментации изображений и создал маски для тренировок.Я храню эти маски в массиве NumPy (значения 1,5, 2, 2,5, 3, 3,5, 4, 4,5, 5).Я попытался экспортировать их в формате .npy, но размер очень большой, и когда я использую matplolib для хранения этих значений в jpg, маска разрушается.Как можно напрямую поместить эти значения массива в виде значений пикселей в изображении.Например, массив numpy имеет размер 300X300, поэтому полученное изображение jpg имеет разрешение 300X300 пикселей.

Answer 1

Здесь две идеи ... вы можете сэкономить некоторое пространство, преобразуя ваши классы в целые числа так:

1.5 -> 1
2 -> 2
2.5 -> 3
...

, что также должно сэкономить некоторое пространство при использовании формата .npy (что я бы порекомендовал).Если вам нужен png для визуального контроля, например, используйте:

import matplotlib as mpl
mpl.image.imsave('test.png',pixel_values)

, и вы можете убедиться в свойствах изображения, что количество пикселей соответствует одному из матрицы pixel_value here.