Save meshes as .obj files. Pytorch3D will not save vertex color for. .obj, please use `save_meshes_as_plys`. Args: files (List[str]): Output .obj file list. meshes (Meshes, optional): Defaults to None.
(files: List[str], meshes: Meshes = None)
| 202 | |
| 203 | |
| 204 | def save_meshes_as_objs(files: List[str], meshes: Meshes = None) -> None: |
| 205 | """Save meshes as .obj files. Pytorch3D will not save vertex color for. |
| 206 | |
| 207 | .obj, please use `save_meshes_as_plys`. |
| 208 | |
| 209 | Args: |
| 210 | files (List[str]): Output .obj file list. |
| 211 | meshes (Meshes, optional): |
| 212 | Defaults to None. |
| 213 | """ |
| 214 | if not isinstance(files, list): |
| 215 | files = [files] |
| 216 | |
| 217 | assert len(files) >= len(meshes), 'Not enough output files.' |
| 218 | |
| 219 | for idx in range(len(meshes)): |
| 220 | prepare_output_path(files[idx], |
| 221 | allowed_suffix=['.obj'], |
| 222 | path_type='file'), 'Please save as .obj files.' |
| 223 | if isinstance(meshes.textures, TexturesUV): |
| 224 | verts_uvs = meshes.textures.verts_uvs_padded()[idx] |
| 225 | faces_uvs = meshes.textures.faces_uvs_padded()[idx] |
| 226 | texture_map = meshes.textures.maps_padded()[idx] |
| 227 | else: |
| 228 | verts_uvs = None |
| 229 | faces_uvs = None |
| 230 | texture_map = None |
| 231 | save_obj(f=files[idx], |
| 232 | verts=meshes.verts_padded()[idx], |
| 233 | faces=meshes.faces_padded()[idx], |
| 234 | verts_uvs=verts_uvs, |
| 235 | faces_uvs=faces_uvs, |
| 236 | texture_map=texture_map) |
no test coverage detected