File size: 1,021 Bytes
434b0b0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | 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)
|