updates for batch processing

1 files changed, 2 insertions, 1 deletions

diff --git a/megapixels/commands/cv/face_vector.py b/megapixels/commands/cv/face_vector.py
index 9a527bc3..cb155d08 100644
--- a/megapixels/commands/cv/face_vector.py
+++ b/megapixels/commands/cv/face_vector.py
@@ -88,7 +88,7 @@ def cli(ctx, opt_fp_out, opt_dir_media, opt_data_store, opt_dataset, opt_size,
 
   # load data
   fp_record = data_store.metadata(types.Metadata.FILE_RECORD)
-  df_record = pd.read_csv(fp_record, dtype={'fn':str}).set_index('index')
+  df_record = pd.read_csv(fp_record, dtype=cfg.FILE_RECORD_DTYPES).set_index('index')
   fp_roi = data_store.metadata(types.Metadata.FACE_ROI)
   df_roi = pd.read_csv(fp_roi).set_index('index')
 
@@ -107,6 +107,7 @@ def cli(ctx, opt_fp_out, opt_dir_media, opt_data_store, opt_dataset, opt_size,
     ds_record = df_record.iloc[record_index]
     fp_im = data_store.face(ds_record.subdir, ds_record.fn, ds_record.ext)
     im = cv.imread(fp_im)
+    im = im_utils.resize(im, width=opt_size[0], height=opt_size[1])
     for roi_index, df_img in df_img_group.iterrows():
       # get bbox
       x, y, w, h = df_img.x, df_img.y, df_img.w, df_img.h