50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# Description: Convert OBJ to NPY
|
||
|
|
# Usage: python3 obj-to-npy.py file.obj
|
||
|
|
# Author: Justin Oros
|
||
|
|
# Source: https://github.com/JustinOros
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
try:
|
||
|
|
import pywavefront
|
||
|
|
except ImportError:
|
||
|
|
print("Please install pywavefront: pip install PyWavefront")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
if len(sys.argv) < 2:
|
||
|
|
print(f"Usage: {sys.argv[0]} <obj-file>")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
obj_path = sys.argv[1]
|
||
|
|
if not os.path.isfile(obj_path):
|
||
|
|
print(f"File not found: {obj_path}")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
base_name = os.path.splitext(os.path.basename(obj_path))[0]
|
||
|
|
npy_path = f"{base_name}.npy"
|
||
|
|
|
||
|
|
print(f"Loading OBJ: {obj_path}")
|
||
|
|
scene = pywavefront.Wavefront(obj_path, create_materials=True, collect_faces=True, parse=True)
|
||
|
|
|
||
|
|
vertices_list = []
|
||
|
|
faces_list = []
|
||
|
|
|
||
|
|
for name, mesh in scene.meshes.items():
|
||
|
|
for material in mesh.materials:
|
||
|
|
for face in mesh.faces:
|
||
|
|
faces_list.append(face)
|
||
|
|
for vertex in scene.vertices:
|
||
|
|
vertices_list.append(vertex)
|
||
|
|
|
||
|
|
vertices_array = np.array(vertices_list, dtype=np.float32)
|
||
|
|
faces_array = np.array(faces_list, dtype=np.int32)
|
||
|
|
|
||
|
|
print(f"Saving binary NPY: {npy_path}")
|
||
|
|
np.save(npy_path, {'vertices': vertices_array, 'faces': faces_array})
|
||
|
|
|
||
|
|
print("Done!")
|
||
|
|
|