Структурирование данных в этом многомерном массиве

Question

Я обрабатываю изображения в python, и для какой-то конкретной цели меня попросили «создать пустые массивы со следующими размерами: 32x3x512x512 (исправляет x цветовых каналов, высоту, ширину)».У меня нет проблем с получением каких-либо данных, но я просто не могу представить структуру массива numpy.Мои лучшие подходы: Вариант 1 , Вариант 2 .И я могу быть даже близко.

Есть идеи, как должен выглядеть массив numpy?

Answer 1

Вы можете буквально взять описание проблемы и использовать его в качестве формы массива:

arr = np.empty((32, 3, 512, 512), 'u1')

Теперь arr[0] является первым патчем (то есть первым изображением) и имеет форму (3, 512, 512)Таким образом, это 24 (3x8) бит на пиксель, а размеры 512 x 512 пикселей.

Некоторые люди могут использовать другой порядок измерений, например (32, 512, 512, 3), но это действительно зависит от обработкиВы в конечном итоге хотите сделать.