diff options
Diffstat (limited to 'megapixels/commands/cv')
| -rw-r--r-- | megapixels/commands/cv/face_landmarks_3d.py (renamed from megapixels/commands/cv/faces_to_3dlm.py) | 4 | ||||
| -rw-r--r-- | megapixels/commands/cv/face_pose.py | 2 | ||||
| -rw-r--r-- | megapixels/commands/cv/face_vector.py | 2 |
3 files changed, 4 insertions, 4 deletions
diff --git a/megapixels/commands/cv/faces_to_3dlm.py b/megapixels/commands/cv/face_landmarks_3d.py index 658d4484..03ef8fc2 100644 --- a/megapixels/commands/cv/faces_to_3dlm.py +++ b/megapixels/commands/cv/face_landmarks_3d.py @@ -1,9 +1,8 @@ """ -Crop images to prepare for training + """ import click -# from PIL import Image, ImageOps, ImageFilter, ImageDraw from app.settings import types from app.utils import click_utils @@ -55,6 +54,7 @@ def cli(ctx, opt_dirs_in, opt_fp_out, opt_ext, opt_size, opt_gpu, opt_slice, # ------------------------------------------------- # init here + log = logger_utils.Logger.getLogger() if not opt_force and Path(opt_fp_out).exists(): diff --git a/megapixels/commands/cv/face_pose.py b/megapixels/commands/cv/face_pose.py index e7ffb7ac..c37d006f 100644 --- a/megapixels/commands/cv/face_pose.py +++ b/megapixels/commands/cv/face_pose.py @@ -95,7 +95,7 @@ def cli(ctx, opt_fp_in, opt_fp_out, opt_dir_media, opt_data_store, opt_dataset, for record_index, df_img_group in tqdm(df_img_groups): # make fp ds_record = df_record.iloc[record_index] - fp_im = data_store.face_image(ds_record.subdir, ds_record.fn, ds_record.ext) + fp_im = data_store.face(ds_record.subdir, ds_record.fn, ds_record.ext) im = cv.imread(fp_im) # get bbox x = df_img_group.x.values[0] diff --git a/megapixels/commands/cv/face_vector.py b/megapixels/commands/cv/face_vector.py index 203f73eb..cd816f9f 100644 --- a/megapixels/commands/cv/face_vector.py +++ b/megapixels/commands/cv/face_vector.py @@ -98,7 +98,7 @@ def cli(ctx, opt_fp_out, opt_dir_media, opt_data_store, opt_dataset, opt_size, ds_file = df_record.loc[roi_index] # locate image meta #ds_file = df_record.loc['index', image_index] # locate image meta - fp_im = data_store.face_image(str(ds_file.subdir), str(ds_file.fn), str(ds_file.ext)) + fp_im = data_store.face(str(ds_file.subdir), str(ds_file.fn), str(ds_file.ext)) im = cv.imread(fp_im) # get bbox x = df_img_group.x.values[0] |