119 lines
5.2 KiB
Python
119 lines
5.2 KiB
Python
#!/usr/bin/env python3
|
|
import os; os.environ.setdefault("PYOPENGL_PLATFORM", "egl")
|
|
from . import config, read
|
|
from ...utils.helpers import make_relative
|
|
from pathlib import Path
|
|
from textwrap import dedent
|
|
import argparse
|
|
|
|
|
|
|
|
def make_parser() -> argparse.ArgumentParser:
|
|
parser = argparse.ArgumentParser(description=dedent("""
|
|
Preprocess the Stanford models. Depends on `download-stanford` having been run.
|
|
"""), formatter_class=argparse.RawTextHelpFormatter)
|
|
|
|
arg = parser.add_argument # brevity
|
|
|
|
arg("objects", nargs="*", default=[],
|
|
help="Which objects to process, defaults to all downloaded")
|
|
arg("--dir", default=str(config.DATA_PATH),
|
|
help=f"The target directory. Default is {make_relative(config.DATA_PATH, Path.cwd()).__str__()!r}")
|
|
arg("--force", action="store_true",
|
|
help="Overwrite existing files")
|
|
arg("--list", action="store_true",
|
|
help="List the downloaded models available for preprocessing")
|
|
arg("--list-pages", type=int, default=None,
|
|
help="List the downloaded models available for preprocessing, paginated into N pages.")
|
|
arg("--page", nargs=2, type=int, default=[0, 1],
|
|
help="Subset of parts to compute. Use to parallelize. (page, total), page is 0 indexed")
|
|
|
|
arg2 = parser.add_argument_group("preprocessing targets").add_argument # brevity
|
|
arg2("--precompute-mesh-sv-scan-clouds", action="store_true",
|
|
help="Compute single-view hit+miss point clouds from 100 synthetic scans.")
|
|
arg2("--precompute-mesh-sv-scan-uvs", action="store_true",
|
|
help="Compute single-view hit+miss UV clouds from 100 synthetic scans.")
|
|
arg2("--precompute-mesh-sphere-scan", action="store_true",
|
|
help="Compute a sphere-view hit+miss cloud cast from n to n unit sphere points.")
|
|
|
|
arg3 = parser.add_argument_group("ray-scan modifiers").add_argument # brevity
|
|
arg3("--n-sphere-points", type=int, default=4000,
|
|
help="The number of unit-sphere points to sample rays from. Final result: n*(n-1).")
|
|
arg3("--compute-miss-distances", action="store_true",
|
|
help="Compute the distance to the nearest hit for each miss in the hit+miss clouds.")
|
|
arg3("--fill-missing-uv-points", action="store_true",
|
|
help="TODO")
|
|
arg3("--no-filter-backhits", action="store_true",
|
|
help="Do not filter scan hits on backside of mesh faces.")
|
|
arg3("--no-unit-sphere", action="store_true",
|
|
help="Do not center the objects to the unit sphere.")
|
|
arg3("--convert-ok", action="store_true",
|
|
help="Allow reusing point clouds for uv clouds and vice versa. (does not account for other hparams)")
|
|
arg3("--debug", action="store_true",
|
|
help="Abort on failiure.")
|
|
|
|
arg5 = parser.add_argument_group("Shared modifiers").add_argument # brevity
|
|
arg5("--scan-resolution", type=int, default=400,
|
|
help="The resolution of the depth map rendered to sample points. Becomes x*x")
|
|
|
|
return parser
|
|
|
|
# entrypoint
|
|
def cli(parser: argparse.ArgumentParser = make_parser()):
|
|
args = parser.parse_args()
|
|
if not any(getattr(args, k) for k in dir(args) if k.startswith("precompute_")) and not (args.list or args.list_pages):
|
|
parser.error("no preprocessing target selected") # exits
|
|
|
|
config.DATA_PATH = Path(args.dir)
|
|
obj_names = args.objects or read.list_object_names()
|
|
|
|
if args.list:
|
|
print(*obj_names, sep="\n")
|
|
parser.exit()
|
|
|
|
if args.list_pages is not None:
|
|
print(*(
|
|
f"--page {i} {args.list_pages} {obj_name}"
|
|
for obj_name in obj_names
|
|
for i in range(args.list_pages)
|
|
), sep="\n")
|
|
parser.exit()
|
|
|
|
if args.precompute_mesh_sv_scan_clouds:
|
|
read.precompute_mesh_scan_point_clouds(
|
|
obj_names,
|
|
compute_miss_distances = args.compute_miss_distances,
|
|
no_filter_backhits = args.no_filter_backhits,
|
|
no_unit_sphere = args.no_unit_sphere,
|
|
convert_ok = args.convert_ok,
|
|
page = args.page,
|
|
force = args.force,
|
|
debug = args.debug,
|
|
)
|
|
if args.precompute_mesh_sv_scan_uvs:
|
|
read.precompute_mesh_scan_uvs(
|
|
obj_names,
|
|
compute_miss_distances = args.compute_miss_distances,
|
|
fill_missing_points = args.fill_missing_uv_points,
|
|
no_filter_backhits = args.no_filter_backhits,
|
|
no_unit_sphere = args.no_unit_sphere,
|
|
convert_ok = args.convert_ok,
|
|
page = args.page,
|
|
force = args.force,
|
|
debug = args.debug,
|
|
)
|
|
if args.precompute_mesh_sphere_scan:
|
|
read.precompute_mesh_sphere_scan(
|
|
obj_names,
|
|
sphere_points = args.n_sphere_points,
|
|
compute_miss_distances = args.compute_miss_distances,
|
|
no_filter_backhits = args.no_filter_backhits,
|
|
no_unit_sphere = args.no_unit_sphere,
|
|
page = args.page,
|
|
force = args.force,
|
|
debug = args.debug,
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
cli()
|