diff options
Diffstat (limited to 'megapixels/commands/cv/face_vector.py')
| -rw-r--r-- | megapixels/commands/cv/face_vector.py | 3 |
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 |