(args)
| 72 | parser.add_argument("--texture_size", type=int, default=4096, help="Baked texture resolution.") |
| 73 | |
| 74 | parser.add_argument("--use_remesh", action="store_true", default=True, help="Remesh input before texturing.") |
| 75 | parser.add_argument("--no_remesh", action="store_false", dest="use_remesh", help="Skip remesh step.") |
| 76 | parser.add_argument("--save_glb", action="store_true", default=True, help="Export textured mesh as .glb.") |
| 77 | parser.add_argument("--no_save_glb", action="store_false", dest="save_glb", help="Keep textured .obj only.") |
| 78 | return parser |
| 79 | |
| 80 | |
| 81 | def process_paint(args): |
| 82 | args_dict = vars(args) |
| 83 | logger.info(args_dict) |
| 84 | |
| 85 | pipeline = PaintPipeline( |
| 86 | model_path=args.model_path, |
| 87 | hy_repo=args.hy_repo, |
| 88 | max_num_view=args.max_num_view, |
| 89 | resolution=args.resolution, |
| 90 | device=args.device, |
| 91 | multiview_cfg_path=args.multiview_cfg_path, |
| 92 | realesrgan_ckpt_path=args.realesrgan_ckpt_path, |
| 93 | custom_pipeline=args.custom_pipeline, |
| 94 | dino_ckpt_path=args.dino_ckpt_path, |
| 95 | render_size=args.render_size, |
| 96 | texture_size=args.texture_size, |
| 97 | ) |
| 98 | result_path = pipeline( |
| 99 | mesh_path=args.mesh_path, |
| 100 | image_path=args.image_path, |
| 101 | save_path=args.save_path, |
| 102 | use_remesh=args.use_remesh, |
no test coverage detected