|
import os
|
|
|
|
import cv2
|
|
import numpy as np
|
|
from PIL import Image
|
|
|
|
selected_indices_old = [
|
|
2311,
|
|
2416,
|
|
2437,
|
|
2460,
|
|
2495,
|
|
2518,
|
|
2520,
|
|
2627,
|
|
4285,
|
|
4315,
|
|
6223,
|
|
6457,
|
|
6597,
|
|
6642,
|
|
6974,
|
|
7054,
|
|
7064,
|
|
7182,
|
|
7303,
|
|
7334,
|
|
7351,
|
|
7368,
|
|
7374,
|
|
7493,
|
|
7503,
|
|
7626,
|
|
8443,
|
|
8562,
|
|
8597,
|
|
8701,
|
|
8817,
|
|
8953,
|
|
11213,
|
|
11261,
|
|
11317,
|
|
11384,
|
|
11600,
|
|
11755,
|
|
11852,
|
|
11891,
|
|
11945,
|
|
12010,
|
|
12354,
|
|
12534,
|
|
12736,
|
|
12880,
|
|
12892,
|
|
13004,
|
|
13323,
|
|
13371,
|
|
13534,
|
|
13575,
|
|
14874,
|
|
14949,
|
|
14977,
|
|
15052,
|
|
15076,
|
|
15291,
|
|
15620,
|
|
15758,
|
|
16309,
|
|
16325,
|
|
16348,
|
|
16390,
|
|
16489,
|
|
16665,
|
|
16891,
|
|
17147,
|
|
17183,
|
|
17488,
|
|
17549,
|
|
17657,
|
|
17932,
|
|
19661,
|
|
20162,
|
|
20200,
|
|
20238,
|
|
20286,
|
|
20432,
|
|
20834,
|
|
20954,
|
|
21015,
|
|
21036,
|
|
21117,
|
|
21299,
|
|
21611,
|
|
21632,
|
|
21649,
|
|
22722,
|
|
22759,
|
|
22873,
|
|
23028,
|
|
23033,
|
|
23082,
|
|
23187,
|
|
23232,
|
|
23302,
|
|
23413,
|
|
23430,
|
|
23446,
|
|
23457,
|
|
23548,
|
|
23636,
|
|
32060,
|
|
32245,
|
|
]
|
|
|
|
selected_indices = list()
|
|
with open('/home/gyalex/Desktop/face_anno.txt', 'r') as f:
|
|
lines = f.readlines()
|
|
for line in lines:
|
|
hh = line.strip().split()
|
|
if len(hh) > 0:
|
|
pid = hh[0].find('.')
|
|
if pid != -1:
|
|
s = hh[0][pid+1:len(hh[0])]
|
|
print(s)
|
|
selected_indices.append(int(s))
|
|
|
|
f.close()
|
|
|
|
dir = '/media/gyalex/Data/face_ldk_dataset/MHC_LightingPreset_Portrait_RT_0_19/MHC_LightingPreset_Portrait_RT_seq_000015'
|
|
|
|
for idx in range(500):
|
|
img = os.path.join(dir, "view_1/MHC_LightingPreset_Portrait_RT_seq_000015_FinalImage_" + str(idx).zfill(4) + ".jpeg")
|
|
lmd = os.path.join(dir, "mesh/mesh_screen" + str(idx+5).zfill(7) + ".npy")
|
|
|
|
img = cv2.imread(img)
|
|
|
|
|
|
|
|
lmd = np.load(lmd)[selected_indices]
|
|
for i in range(lmd.shape[0]):
|
|
p = lmd[i]
|
|
x, y = round(float(p[0])), round(float(p[1]))
|
|
print(p)
|
|
cv2.circle(img, (x, y), 2, (0, 0, 255), -1)
|
|
|
|
cv2.imshow('win', img)
|
|
cv2.waitKey(0) |
