search

K0nkere / DL_Dice-detection-project

DnD dice detection with CNN and transfer learning / Project for ML Bookcamp
0 stars 0 forks source link

PIL #1

Open K0nkere opened 1 year ago

K0nkere commented 1 year ago

Создать PIL image из numpy.ndarray

msk = labeled_mask[:, :, 1]
size = labeled_mask.shape[0]

blank = np.zeros((size, size, 1), dtype='int')
msk = np.array(msk[:, :, np.newaxis], dtype='int') * 255

msk = np.concatenate((msk, blank, blank), axis=-1)
msk_image = Image.fromarray(msk, mode="RGB")