YuliangXiu / ECON

[CVPR'23, Highlight] ECON: Explicit Clothed humans Optimized via Normal integration
https://xiuyuliang.cn/econ
Other
1.08k stars 106 forks source link

Check if meshes are empty #74

Closed KernelA closed 1 year ago

KernelA commented 1 year ago

This image lead to error, because hand or face mesh can be empty. This pull request contains checks if any of hand or face mesh is empty.

Doctor

python -m apps.infer -cfg ./configs/econ.yaml -in_dir ./examples -out_dir ./results -novis
Traceback (most recent call last):
  File ".../envs/.../lib/python3.9/runpy.py", line 197, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File ".../envs/.../lib/python3.9/runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File ".../ECON/apps/infer.py", line 595, in <module>
    BNI_object.F_B_trimesh = part_removal(
  File ".../ECON/lib/dataset/mesh_util.py", line 270, in part_removal
    (part_dist, _) = part_extractor.query(torch.tensor(full_mesh.vertices).unsqueeze(0).to(device))
  File ".../ECON/lib/dataset/PointFeat.py", line 39, in query
    residues, pts_ind = point_mesh_distance(self.mesh, Pointclouds(points), weighted=False)
  File ".../ECON/lib/dataset/Evaluator.py", line 213, in point_mesh_distance
    verts_packed = meshes.verts_packed()
  File ".../envs/.../lib/python3.9/site-packages/pytorch3d/structures/meshes.py", line 563, in verts_packed
    self._compute_packed()
  File ".../envs/.../lib/python3.9/site-packages/pytorch3d/structures/meshes.py", line 974, in _compute_packed
    verts_list = self.verts_list()
  File ".../envs/.../lib/python3.9/site-packages/pytorch3d/structures/meshes.py", line 535, in verts_list
    self._verts_list = struct_utils.padded_to_list(
  File ".../envs/.../lib/python3.9/site-packages/pytorch3d/structures/utils.py", line 106, in padded_to_list
    raise ValueError("Split size must be of same length as inputs first dimension")
ValueError: Split size must be of same length as inputs first dimension
YuliangXiu commented 1 year ago

@KernelA Thanks, commit is merged.