diff options
Diffstat (limited to 'megapixels/commands/demo/face_search.py')
| -rw-r--r-- | megapixels/commands/demo/face_search.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/megapixels/commands/demo/face_search.py b/megapixels/commands/demo/face_search.py index d50f5c73..f551cafd 100644 --- a/megapixels/commands/demo/face_search.py +++ b/megapixels/commands/demo/face_search.py @@ -39,6 +39,7 @@ def cli(ctx, opt_fp_in, opt_data_store, opt_dataset, opt_results, opt_gpu): import cv2 as cv from tqdm import tqdm import imutils + from PIL import Image, ImageOps from app.utils import file_utils, im_utils, display_utils, draw_utils from app.models.data_store import DataStore @@ -49,9 +50,10 @@ def cli(ctx, opt_fp_in, opt_data_store, opt_dataset, opt_results, opt_gpu): # init dataset dataset = Dataset(opt_data_store, opt_dataset) - dataset.load_face_vectors() - dataset.load_records() - # dataset.load_identities() + dataset.load_metadata(types.Metadata.FILE_RECORD) + dataset.load_metadata(types.Metadata.FACE_VECTOR) + dataset.load_metadata(types.Metadata.FACE_ROI) + # dataset.load_metadata(types.Metadata.IDENTITY) # init face detection detector = face_detector.DetectorCVDNN() @@ -86,6 +88,9 @@ def cli(ctx, opt_fp_in, opt_data_store, opt_dataset, opt_results, opt_gpu): image_record.summarize() log.info(f'{image_record.filepath}') im_match = cv.imread(image_record.filepath) + + im_match_pil = Image.open(image_record.filepath).convert('RGB') + # bbox = ims_match.append(im_match) # make montages of most similar faces |