Problems with training.py

[geoyi]

Problems with training.py

[Geoyi]

哈哈，谢谢你的赞赏, @chloe-hampton。

The error comes from the labels. Label Maker creates 3 color channel labels for preview purpose, and you have to reshape the numpy array from (256, 256, 3) to (256, 256). I added utility script to convert it, see the script in this jupyter notebook I prepared for a conference.

In short, you can directly do:

from os import makedirs, path as op
from PIL import Image
import numpy as np

labels = np.load('labels.npz')

for label in labels.files:
    new_label_dir = op.join(os.getcwd(), "new_labels")
    if not op.isdir(new_label_dir):
        makedirs(new_label_dir)
    img = Image.fromarray(labels[label])
    print('Writing {}'.format(label))
    img.save('{}/{}.png'.format(new_label_dir, label))