|
|
import json
|
|
|
import os
|
|
|
import numpy as np
|
|
|
from glob import glob
|
|
|
|
|
|
|
|
|
def export_expression_json(flame_param_dir, output_path):
|
|
|
all_flames_npz = glob(os.path.join(flame_param_dir,'*.npz'))
|
|
|
all_flames_npz = sorted(all_flames_npz)
|
|
|
output_json = None
|
|
|
for flame_path in all_flames_npz:
|
|
|
flame_param = dict(np.load(flame_path), allow_pickle=True)
|
|
|
expr = flame_param['expr']
|
|
|
expr_dict = {f"expr{i}": float(item) for i, item in enumerate(expr.squeeze(0))}
|
|
|
flame_param['expr'] = expr_dict
|
|
|
if(output_json is None):
|
|
|
output_json = flame_param
|
|
|
try:
|
|
|
del(output_json['static_offset'])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
del(output_json['allow_pickle'])
|
|
|
except:
|
|
|
pass
|
|
|
else:
|
|
|
for key in output_json:
|
|
|
if('shape' in key): continue
|
|
|
if 'expr' in key:
|
|
|
if isinstance(output_json.get(key, None), dict):
|
|
|
output_json[key] = [flame_param[key]]
|
|
|
else:
|
|
|
output_json[key].append(flame_param[key])
|
|
|
pass
|
|
|
else:
|
|
|
output_json[key] = np.concatenate([output_json[key], flame_param[key]], axis=0)
|
|
|
|
|
|
for key in output_json:
|
|
|
if key != "expr":
|
|
|
output_json[key] = output_json[key].tolist()
|
|
|
|
|
|
with open(output_path,'w') as f:
|
|
|
json.dump(output_json,f)
|
|
|
|
|
|
print("Save flame params into:", output_path)
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
root = './export/nice/'
|
|
|
|
|
|
|
|
|
motion_lst = [l.strip() for l in open("./fd_lst.txt").readlines()]
|
|
|
for motion in motion_lst:
|
|
|
root = os.path.join('./export/', motion)
|
|
|
input_dir = os.path.join(root, "flame_param")
|
|
|
export_expression_json(input_dir, os.path.join(root, 'flame_params.json'))
|
|
|
|
