From 4bcb82c0f295d79d3d247252e7e98b2d986ae821 Mon Sep 17 00:00:00 2001 From: adamhrv Date: Sun, 6 Jan 2019 17:16:18 +0100 Subject: externalize drawing, cleanup --- megapixels/commands/cv/face_vector.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'megapixels/commands/cv/face_vector.py') diff --git a/megapixels/commands/cv/face_vector.py b/megapixels/commands/cv/face_vector.py index 7c03205c..9251c053 100644 --- a/megapixels/commands/cv/face_vector.py +++ b/megapixels/commands/cv/face_vector.py @@ -103,15 +103,17 @@ def cli(ctx, opt_fp_out, opt_dir_media, opt_data_store, opt_dataset, opt_size, # get face vector bbox_dim = BBox.from_xywh(x, y, w, h).to_dim(dim) # convert to int real dimensions # compute vec - # padding=opt_padding not yet implemented in 19.16 but merged in master + # padding=opt_padding not yet implemented in dlib===19.16 but merged in master vec = facerec.vec(im, bbox_dim, jitters=opt_jitters) vec_str = ','.join([repr(x) for x in vec]) # convert to string for CSV vecs.append( {'roi_index': roi_index, 'record_index': record_index, 'vec': vec_str}) - # create dataframe + # create DataFrame and save to CSV df = pd.DataFrame.from_dict(vecs) df.index.name = 'index' - # save CSV file_utils.mkdirs(fp_out) - df.to_csv(fp_out) \ No newline at end of file + df.to_csv(fp_out) + + # save script + file_utils.write_text(' '.join(sys.argv), '{}.sh'.format(fp_out)) \ No newline at end of file -- cgit v1.2.3-70-g09d2