aboutsummaryrefslogtreecommitdiff
path: root/src/main.rs
diff options
context:
space:
mode:
authorLia Lenckowski <lialenck@protonmail.com>2023-09-05 21:41:17 +0200
committerLia Lenckowski <lialenck@protonmail.com>2023-09-05 21:41:17 +0200
commit3fe8109102083ef32da9e11eef5dc45dba530333 (patch)
tree0480ee88e9879a13dfb8fedb4d0514fab42f7e31 /src/main.rs
parent7286cf533a1efaae30189da608d0025ae63d0fa2 (diff)
downloadembeddings-sort-3fe8109102083ef32da9e11eef5dc45dba530333.tar
embeddings-sort-3fe8109102083ef32da9e11eef5dc45dba530333.tar.bz2
embeddings-sort-3fe8109102083ef32da9e11eef5dc45dba530333.tar.zst
actually use 'Hue' in the HueEmbedder, add ColorEmbedder, parallelize image embedding
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/main.rs b/src/main.rs
index f34413a..a28885d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -14,6 +14,7 @@ mod embedders;
enum Embedder {
Brightness,
Hue,
+ Color,
}
#[derive(Debug, Parser)]
@@ -124,6 +125,7 @@ fn main() -> Result<(), String> {
let tsp_path = match args.embedder {
Embedder::Brightness => process_embedder(BrightnessEmbedder, args),
Embedder::Hue => process_embedder(HueEmbedder, args),
+ Embedder::Color => process_embedder(ColorEmbedder, args),
}?;
for p in tsp_path {