55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
|
"""Aling body models s.t. they stand exactly on the plane y=0 and save as a new data"""
|
||
|
|
import igl
|
||
|
|
import numpy as np
|
||
|
|
from pathlib import Path
|
||
|
|
import trimesh
|
||
|
|
|
||
|
|
from pygarment.data_config import Properties
|
||
|
|
|
||
|
|
def load_mesh(path):
|
||
|
|
v, f = igl.read_triangle_mesh(str(path))
|
||
|
|
return v, f.flatten(), f
|
||
|
|
|
||
|
|
def get_shift_param(body_vertices):
|
||
|
|
v_body_arr = np.array(body_vertices)
|
||
|
|
min_y = (min(v_body_arr[:, 1]))
|
||
|
|
if min_y < 0:
|
||
|
|
return abs(min_y)
|
||
|
|
return 0.0
|
||
|
|
|
||
|
|
def save_mesh(path, v, f):
|
||
|
|
igl.write_triangle_mesh(str(path), v=v, f=f, force_ascii=False)
|
||
|
|
|
||
|
|
|
||
|
|
def process_body(path_in, path_out):
|
||
|
|
|
||
|
|
body_vertices, _, body_faces = load_mesh(path_in) # self.paths.in_body_obj)
|
||
|
|
shift_y = get_shift_param(body_vertices)
|
||
|
|
|
||
|
|
# body_vertices = body_vertices * b_scale
|
||
|
|
if shift_y:
|
||
|
|
body_vertices[:, 1] = body_vertices[:, 1] + shift_y
|
||
|
|
|
||
|
|
save_mesh(path_out, body_vertices, body_faces)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
|
||
|
|
system_paths = Properties('./system.json')
|
||
|
|
|
||
|
|
# body_folder_path = Path(system_paths['body_samples_path']) / 'body_shapes_and_measures_2023-12-30'
|
||
|
|
# body_objs_path = body_folder_path / 'meshes'
|
||
|
|
body_objs_path = Path('./assets/bodies')
|
||
|
|
# out_path = body_folder_path / 'meshes_aligned'
|
||
|
|
out_path = Path('./assets/bodies_aligned')
|
||
|
|
out_path.mkdir(parents=True, exist_ok=True)
|
||
|
|
|
||
|
|
# loop over all meshes
|
||
|
|
for file in body_objs_path.iterdir():
|
||
|
|
if '.obj' in file.name:
|
||
|
|
process_body(file, out_path / file.name)
|
||
|
|
|
||
|
|
print(file.name)
|
||
|
|
|
||
|
|
|
||
|
|
|