| 54 | |
| 55 | |
| 56 | def add_sampling_options(parser): |
| 57 | group = parser.add_argument_group("sampling") |
| 58 | group.add_argument("--n_samples", type=int, default=1, help="number of samples") |
| 59 | group.add_argument("--input", type=str, default=None, help="input folder") |
| 60 | group.add_argument("--output", type=str, default="results", help="output folder") |
| 61 | group.add_argument("--resize", default=(1, 1, 1), type=float, nargs=3, help="resize factor") |
| 62 | group.add_argument("--use_ddim", type=str2bool, default=False, help="use ddim") |
| 63 | group.add_argument("--timestep_respacing", type=str, default="", help="timestep respacing") |
| 64 | group.add_argument("--app", type=str, default="generate", help="") |
| 65 | |
| 66 | group.add_argument("--reso", type=int, default=256, help="decoding volume resolution") |
| 67 | group.add_argument("--n_faces", type=int, default=10000, help="number of simplified mesh faces") |
| 68 | group.add_argument("--texreso", type=int, default=2048, help="texture resolution") |
| 69 | group.add_argument("--vox", action='store_true', help="") |
| 70 | |
| 71 | group.add_argument("--copy_mtl", type=str2bool, default=True, help="copy mtl file") |
| 72 | group.add_argument("--file_format", type=str, default="obj", choices=["obj", "glb"], help="file format") |
| 73 | |
| 74 | |
| 75 | def diffusion_defaults(): |