| 6 | from pytorch3d.io import load_obj, save_obj |
| 7 | |
| 8 | def parse_args(): |
| 9 | parser = argparse.ArgumentParser('Model') |
| 10 | parser.add_argument('--src_shape', type=str, default="../data/demo/104256", help='source shape') |
| 11 | parser.add_argument('--tar_shape', type=str, default="../data/demo/834af6", help='target shape') |
| 12 | parser.add_argument('--checkpoint', type=str, default="../checkpoints/chair_15.pth", help='checkpoint path') |
| 13 | parser.add_argument('--save_dir', type=str, default="104256_meta_handle", help='save dir') |
| 14 | parser.add_argument('--num_basis', type=int, default=15, help='number of basis vectors') |
| 15 | return parser.parse_args() |
| 16 | |
| 17 | opt = parse_args() |
| 18 | net = network.model(opt.num_basis).cuda() |