Function extract_data

boss_input.py:61–66

(path)

def extract_data(path):
    images, labels = traverse_dir(path)
    images = np.array(images)
    labels = np.array([0 if label.endswith('boss') else 1 for label in labels])

    return images, labels