mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-10 18:17:31 +08:00
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
"""
|
|
Script to texture an existing mesh file.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
import tyro
|
|
from rich.console import Console
|
|
from typing_extensions import Literal
|
|
|
|
from nerfstudio.exporter import texture_utils
|
|
from nerfstudio.exporter.exporter_utils import get_mesh_from_filename
|
|
from nerfstudio.utils.eval_utils import eval_setup
|
|
|
|
CONSOLE = Console(width=120)
|
|
|
|
|
|
@dataclass
|
|
class TextureMesh:
|
|
"""
|
|
Export a textured mesh with color computed from the NeRF.
|
|
"""
|
|
|
|
load_config: Path
|
|
"""Path to the config YAML file."""
|
|
output_dir: Path
|
|
"""Path to the output directory."""
|
|
input_mesh_filename: Path
|
|
"""Mesh filename to texture."""
|
|
px_per_uv_triangle: int = 4
|
|
"""Number of pixels per UV square."""
|
|
unwrap_method: Literal["xatlas", "custom"] = "xatlas"
|
|
"""The method to use for unwrapping the mesh."""
|
|
num_pixels_per_side: int = 2048
|
|
"""If using xatlas for unwrapping, the pixels per side of the texture image."""
|
|
target_num_faces: Optional[int] = 50000
|
|
"""Target number of faces for the mesh to texture."""
|
|
|
|
def main(self) -> None:
|
|
"""Export textured mesh"""
|
|
# pylint: disable=too-many-statements
|
|
|
|
if not self.output_dir.exists():
|
|
self.output_dir.mkdir(parents=True)
|
|
|
|
# load the Mesh
|
|
mesh = get_mesh_from_filename(str(self.input_mesh_filename), target_num_faces=self.target_num_faces)
|
|
|
|
# load the Pipeline
|
|
_, pipeline, _ = eval_setup(self.load_config, test_mode="inference")
|
|
|
|
# texture the mesh with NeRF and export to a mesh.obj file
|
|
# and a material and texture file
|
|
texture_utils.export_textured_mesh(
|
|
mesh=mesh,
|
|
pipeline=pipeline,
|
|
output_dir=self.output_dir,
|
|
px_per_uv_triangle=self.px_per_uv_triangle,
|
|
unwrap_method=self.unwrap_method,
|
|
num_pixels_per_side=self.num_pixels_per_side,
|
|
)
|
|
|
|
|
|
def entrypoint():
|
|
"""Entrypoint for use with pyproject scripts."""
|
|
tyro.extras.set_accent_color("bright_yellow")
|
|
tyro.cli(tyro.conf.FlagConversionOff[TextureMesh]).main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
entrypoint()
|
|
|
|
# For sphinx docs
|
|
get_parser_fn = lambda: tyro.extras.get_parser(TextureMesh) # noqa
|