mirror of
https://github.com/deepinsight/insightface.git
synced 2026-05-19 15:41:33 +00:00
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
|
|
import sys
|
|
import os
|
|
import numpy as np
|
|
|
|
input_dir = sys.argv[1]
|
|
targets = sys.argv[2]
|
|
targets = targets.strip().split(',')
|
|
lmap = {}
|
|
|
|
for ds in targets:
|
|
#image_dir = os.path.join(input_dir, ds)
|
|
lmk_file = os.path.join(input_dir, "%s_lmk"%(ds))
|
|
if not os.path.exists(lmk_file):
|
|
lmk_file = os.path.join(input_dir, "%s_lmk.txt"%(ds))
|
|
if not os.path.exists(lmk_file):
|
|
continue
|
|
#print(ds)
|
|
idx = 0
|
|
for line in open(lmk_file, 'r'):
|
|
idx+=1
|
|
vec = line.strip().split(' ')
|
|
assert len(vec)==12 or len(vec)==11
|
|
image_file = os.path.join(input_dir, vec[0])
|
|
assert image_file.endswith('.jpg')
|
|
vlabel = -1 #test mode
|
|
if len(vec)==12:
|
|
label = int(vec[1])
|
|
if label in lmap:
|
|
vlabel = lmap[label]
|
|
else:
|
|
vlabel = len(lmap)
|
|
lmap[label] = vlabel
|
|
lmk = np.array([float(x) for x in vec[2:]], dtype=np.float32)
|
|
else:
|
|
lmk = np.array([float(x) for x in vec[1:]], dtype=np.float32)
|
|
lmk = lmk.reshape( (5,2) ).T
|
|
lmk_str = "\t".join( [str(x) for x in lmk.flatten()] )
|
|
print("0\t%s\t%d\t0\t0\t0\t0\t%s"%(image_file, vlabel, lmk_str))
|
|
#if idx>10:
|
|
# break
|
|
|