pixel_sampling.py 文件源码

def learn_color_clusters():
    samples = np.zeros((0, 3))
    cnt = 0
    with open('train_list') as f:
        for line in f:
            line = line[:-1]
            image = cv2.imread(line)
            image = cv2.resize(image, (100, 100))
            image = cv2.cvtColor(image, cv2.COLOR_BGR2Lab)

            points = image.reshape((-1, 3))
            np.random.permutation(points.shape[0])
            samples = np.vstack([samples, points[:50]])

            print(samples.shape)
            cnt = cnt + 1
            if cnt % 10000 == 0:
                break

    km = cluster.KMeans(n_clusters=50, n_jobs=-1)
    km.fit(samples)
    np.save('lab_clusters.npy', km.cluster_centers_)
    return

#learn_color_clusters()