engine/NormalEstimator/Sapiens/interface.py

import os
import sys

sys.path.append("./")

SAPIENS_WEIGHT = "/mnt/workspaces/codes/nips-2024/CHAMP/pretrained_models/sapiens/normal/sapiens_1b_normal_render_people_epoch_115_torchscript.pt2"


def matting_process_imgs(path):

    src_path = path
    target = os.path.join(src_path, "masks")

    cmd = f"python ./human_generate_system/engineer/SegmentAPI/SAM.py -i {path} -o {target} --mask --wo_super_reso"
    print(cmd)

    os.system(cmd)


def interface(folder, mask_folder, save_folder):

    cmd = f"python ./engine/NormalEstimator/Sapiens/core/vis_normal.py \
            {SAPIENS_WEIGHT} \
            --input {folder} \
            --seg_dir {mask_folder} \
            --batch-size=1 --output-root={save_folder}"

    print(cmd)

    os.system(cmd)


if __name__ == "__main__":
    folders = [
        "/mnt/workspaces/codes/nips-2024/Online_816/Human3DGS/gradio_examples/mv_texture_smpl/09F273EE-9E5E-4DAF-BB96-8E0923E6663D-2874-000001831D0983D7/"
    ]
    for folder in folders:
        interface(folder)