From a52dc6f8edc4e5d00e7c05e34c6c6fe6252ec2bd Mon Sep 17 00:00:00 2001 From: adamhrv Date: Sun, 16 Dec 2018 01:03:07 +0100 Subject: add Dataset model, stub face_*analsysi --- megapixels/commands/cv/cluster.py | 47 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 47 insertions(+) create mode 100644 megapixels/commands/cv/cluster.py (limited to 'megapixels/commands/cv') diff --git a/megapixels/commands/cv/cluster.py b/megapixels/commands/cv/cluster.py new file mode 100644 index 00000000..94334133 --- /dev/null +++ b/megapixels/commands/cv/cluster.py @@ -0,0 +1,47 @@ +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 -- cgit v1.2.3-70-g09d2