updates for batch processing

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