paulpanwang/PartCrafter

1import os2import trimesh3import numpy as np4import argparse5import json6 7from src.utils.data_utils import normalize_mesh8from src.utils.render_utils import render_single_view9 10RADIUS = 411IMAGE_SIZE = (2048, 2048)12LIGHT_INTENSITY = 2.513NUM_ENV_LIGHTS = 3614 15if __name__ == '__main__':16    parser = argparse.ArgumentParser()17    parser.add_argument('--input', type=str, default='assets/objects/scissors.glb')18    parser.add_argument('--output', type=str, default='preprocessed_data')19    args = parser.parse_args()20 21    input_path = args.input22    output_path = args.output23 24    assert os.path.exists(input_path), f'{input_path} does not exist'25 26    mesh_name = os.path.basename(input_path).split('.')[0]27    output_path = os.path.join(output_path, mesh_name)28    if not os.path.exists(output_path):29        os.makedirs(output_path)30 31    mesh = normalize_mesh(trimesh.load(input_path, process=False))32    mesh = mesh.to_geometry()33    image = render_single_view(34        mesh,35        radius=RADIUS,36        image_size=IMAGE_SIZE,37        light_intensity=LIGHT_INTENSITY,38        num_env_lights=NUM_ENV_LIGHTS,39        return_type='pil'40    )41    image.save(os.path.join(output_path, f'rendering.png'))