создание пустого 3-канального изображения и добавление значений пикселей

Question

Я пытаюсь создать пустое изображение с 3 каналами:

heatmap  = np.zeros_like((416,416,3))

Теперь я хочу, чтобы он действовал как псевдо-тепловая карта, поэтому я создал поле

box = [0,0,25,25]

А теперь увеличьте значение пикселей в этом поле на пустом изображении на 1

while( box[0] < box[2]):
    y_axis = int(box[1])
    while(y_axis < box[3]):
            heatmap[box[0]][y_axis] += 1
            y_axis  = y_axis+1
    box[0] = box[0]+1

Используя это, я получаю цвета, но "полные" цвета, мне нужен прозрачный цвет, чтобы я мог поместить эту "псевдо-тепловую карту" поверх изображения.

Как мне этого достичь?

Answer 1

Для создания прозрачного изображения вам нужно 4-канальное изображение, 4-й канал представляет альфа-канал. Чтобы создать прозрачное изображение, вы можете игнорировать значения RGB и напрямую установить альфа-канал равным 0. Таким образом, вы можете использовать это:

heatmap  = np.zeros_like((416,416,4),dtype=np.uint8)