summaryrefslogtreecommitdiff
path: root/megapixels/commands/cv/cluster.py
diff options
context:
space:
mode:
authoradamhrv <adam@ahprojects.com>2019-01-18 11:00:18 +0100
committeradamhrv <adam@ahprojects.com>2019-01-18 11:00:18 +0100
commite06af50389f849be0bfe4fa97d39f4519ef2c711 (patch)
tree49755b51e1b8b1f8031e5483333570a8e9951272 /megapixels/commands/cv/cluster.py
parent03ad11fb2a3dcd425d50167b15d72d4e0ef536a2 (diff)
change to cli_proc
Diffstat (limited to 'megapixels/commands/cv/cluster.py')
-rw-r--r--megapixels/commands/cv/cluster.py47
1 files changed, 0 insertions, 47 deletions
diff --git a/megapixels/commands/cv/cluster.py b/megapixels/commands/cv/cluster.py
deleted file mode 100644
index 419091a0..00000000
--- a/megapixels/commands/cv/cluster.py
+++ /dev/null
@@ -1,47 +0,0 @@
-import click
-
-from app.settings import types
-from app.utils import click_utils
-from app.settings import app_cfg as cfg
-from app.utils.logger_utils import Logger
-
-@click.command()
-@click.option('--data_store', 'opt_data_store',
- type=cfg.DataStoreVar,
- default=click_utils.get_default(types.DataStore.NAS),
- show_default=True,
- help=click_utils.show_help(types.Dataset))
-@click.option('--dataset', 'opt_dataset',
- type=cfg.DatasetVar,
- required=True,
- show_default=True,
- help=click_utils.show_help(types.Dataset))
-@click.option('--metadata', 'opt_metadata', required=True,
- type=cfg.MetadataVar,
- show_default=True,
- help=click_utils.show_help(types.Metadata))
-@click.pass_context
-def cli(ctx, opt_data_store, opt_dataset, opt_metadata):
- """Display image info"""
-
- # cluster the embeddings
- print("[INFO] clustering...")
- clt = DBSCAN(metric="euclidean", n_jobs=args["jobs"])
- clt.fit(encodings)
-
- # determine the total number of unique faces found in the dataset
- labelIDs = np.unique(clt.labels_)
- numUniqueFaces = len(np.where(labelIDs > -1)[0])
- print("[INFO] # unique faces: {}".format(numUniqueFaces))
- # load and display image
- im = cv.imread(fp_im)
- cv.imshow('', im)
-
- while True:
- k = cv.waitKey(1) & 0xFF
- if k == 27 or k == ord('q'): # ESC
- cv.destroyAllWindows()
- sys.exit()
- elif k != 255:
- # any key to continue
- break \ No newline at end of file