Function parse_args

src/demo_target_driven_deform.py:8–15

()

from pytorch3d.io import load_obj, save_obj

def parse_args():
    parser = argparse.ArgumentParser('Model')
    parser.add_argument('--src_shape', type=str, default="../data/demo/2e0beb", help='source shape')
    parser.add_argument('--tar_shape', type=str, default="../data/demo/834af6", help='target shape')
    parser.add_argument('--checkpoint', type=str, default="../checkpoints/chair_15.pth", help='checkpoint path') 
    parser.add_argument('--save_name', type=str, default="2e0beb-834af6-def.obj", help='save name')
    parser.add_argument('--num_basis', type=int, default=15, help='number of basis vectors')
    return parser.parse_args()
    
opt = parse_args()
net = network.model(opt.num_basis).cuda()