From b09d6a0a2d9887b5560a3b49cb8a603a33ec4828 Mon Sep 17 00:00:00 2001 From: Chandler J Date: Sat, 6 Apr 2024 20:52:13 -0600 Subject: improved color picker --- src/color_engine.py | 2 -- 1 file changed, 2 deletions(-) (limited to 'src/color_engine.py') diff --git a/src/color_engine.py b/src/color_engine.py index 0e43475..c24450b 100644 --- a/src/color_engine.py +++ b/src/color_engine.py @@ -13,8 +13,6 @@ def grabColors(img_path: str, num_colors: int) -> list: # scale image down by factor of 10 to decrease computation time dim = (int(len(img[0])/10), int(len(img)/10)) img = cv.resize(img, dim, interpolation= cv.INTER_AREA) - # TODO: implement KMeans clustering from scratch to - # improve program modularity clt = KMeans(n_clusters=num_colors, n_init='auto') clt.fit(img.reshape(-1, 3)) return clt.cluster_centers_ -- cgit v1.2.3