fix

ternaus · Jul 29, 2020 · 9f3f934 · 9f3f934
1 parent 7b6e5bc
commit 9f3f934
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/retinaface/inference.py b/retinaface/inference.py
@@ -229,7 +229,7 @@ def main():
 
     file_paths = []
 
-    for regexp in ["*.jpg", "*.png", "*.jpeg", "*.JPG", ""]:
+    for regexp in ["*.jpg", "*.png", "*.jpeg", "*.JPG"]:
         file_paths += sorted([x for x in tqdm(args.input_path.rglob(regexp))])
 
     dataset = InferenceDataset(file_paths, max_size=args.max_size, transform=from_dict(hparams["test_aug"]))