fix dir not found in test

lcp29 · Jan 10, 2025 · a5f6252 · a5f6252
1 parent 637e270
commit a5f6252
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/test/performance_test.py b/test/performance_test.py
@@ -43,6 +43,9 @@ def gen_rays(cam_mat, w, h, f):
     .broadcast_to(ray_dirs.shape)
 )
 model_path = os.path.dirname(os.path.abspath(__file__)) + '/models/iscv2.obj'
+model_dir = os.path.dirname(os.path.abspath(__file__)) + '/models'
+if not os.path.exists(model_dir):
+    os.makedirs(model_dir)
 if not os.path.exists(model_path):
     os.system(f'wget https://file.fomal.host/iscv2.obj -O {model_path}')
 mesh = trimesh.load(model_path, force='mesh')