added 'sort by vggface': sorting by face similarity using VGGFace model.

Requires 4GB+ VRAM and internet connection for the first run.
2019-10-23 15:06:39 +04:00
parent 0d3b25812d
commit 734d97d729
8 changed files with 186 additions and 43 deletions
@@ -37,6 +37,10 @@ def extract_vggface2_dataset(input_dir, device_args={} ):
        
        cur_input_path = input_path / dir_name
        cur_output_path = output_path / dir_name
+        
+        l = len(Path_utils.get_image_paths(cur_input_path))
+        if l < 250 or l > 350:
+            continue

        io.log_info (f"Processing: {str(cur_input_path)} ")