mirror of
https://github.com/deepinsight/insightface.git
synced 2026-05-20 00:10:28 +00:00
14 lines
210 B
Python
14 lines
210 B
Python
|
|
import sys
|
|
import os
|
|
import face_image
|
|
|
|
input_dir = sys.argv[1]
|
|
|
|
dataset = face_image.get_dataset_common(input_dir, 2)
|
|
|
|
for item in dataset:
|
|
print("%d\t%s\t%d" % (1, item.image_path, int(item.classname)))
|
|
|
|
|