|
""" |
|
Processes a directory containing *.jpg/png and outputs crops and poses. |
|
""" |
|
import glob |
|
import os |
|
import subprocess |
|
import argparse |
|
parser = argparse.ArgumentParser() |
|
parser.add_argument('--input_dir', default='/media/data6/ericryanchan/mafu/Deep3DFaceRecon_pytorch/test_images') |
|
parser.add_argument('--gpu', default=0) |
|
args = parser.parse_args() |
|
|
|
|
|
|
|
|
|
print("Computing facial landmarks for model...") |
|
cmd = "python batch_mtcnn.py" |
|
input_flag = " --in_root " + args.input_dir |
|
cmd += input_flag |
|
subprocess.run([cmd], shell=True, check=True) |
|
|
|
|
|
|
|
print("Running model inference...") |
|
cmd = "python test.py" |
|
input_flag = " --img_folder=" + args.input_dir |
|
gpu_flag = " --gpu_ids=" + str(args.gpu) |
|
model_name_flag = " --name=pretrained" |
|
model_file_flag = " --epoch=20 " |
|
cmd += input_flag + gpu_flag + model_name_flag + model_file_flag |
|
subprocess.run([cmd], shell=True, check=True) |
|
|
|
|
|
print("Processing final crops...") |
|
cmd = "python crop_images.py" |
|
input_flag = " --indir " + args.input_dir |
|
output_flag = " --outdir " + os.path.join(args.input_dir, 'cropped_images') |
|
cmd += input_flag + output_flag |
|
subprocess.run([cmd], shell=True, check=True) |
|
|
|
|
|
print("Processing final poses...") |
|
cmd = "python 3dface2idr.py" |
|
input_flag = " --in_root " + os.path.join(args.input_dir, "epoch_20_000000") |
|
output_flag = " --out_root " + os.path.join(args.input_dir, "cropped_images") |
|
|
|
cmd += input_flag + output_flag |
|
subprocess.run([cmd], shell=True, check=True) |
