From c72e2d0e7c7421e03e4cead9ede388b71d4cb836 Mon Sep 17 00:00:00 2001 From: Prajwal Date: Sat, 3 Oct 2020 10:51:13 +0530 Subject: [PATCH] Force args.face to be a valid file --- inference.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/inference.py b/inference.py index 9aeee60..58cc69f 100644 --- a/inference.py +++ b/inference.py @@ -179,9 +179,7 @@ def load_model(path): def main(): if not os.path.isfile(args.face): - fnames = list(glob(os.path.join(args.face, '*.jpg'))) - sorted_fnames = sorted(fnames, key=lambda f: int(os.path.basename(f).split('.')[0])) - full_frames = [cv2.imread(f) for f in sorted_fnames] + raise ValueError('--face argument must be a valid path to video/image file') elif args.face.split('.')[1] in ['jpg', 'png', 'jpeg']: full_frames = [cv2.imread(args.face)]