From dff06bce4977aa1cfbfc21a9312edb8d0a16f71c Mon Sep 17 00:00:00 2001
From: Ethan Yang <thatprogrammer1@gmail.com>
Date: Thu, 20 Feb 2025 17:18:34 -0500
Subject: [PATCH] fix infinite loop if no models found

---
 datareader.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/datareader.py b/datareader.py
index b7a660681..cbb0ffbd2 100644
--- a/datareader.py
+++ b/datareader.py
@@ -416,12 +416,15 @@ def __init__(self, base_dir='/mnt/9a72c439-d0a7-45e8-8d20-d7a235d02763/DATASET/L
 
   def get_gt_mesh_file(self, ob_id):
     root = self.base_dir
-    while 1:
+    mesh_dir = None
+    while root!=os.path.abspath('/'):
       if os.path.exists(f'{root}/lm_models'):
         mesh_dir = f'{root}/lm_models/models/obj_{ob_id:06d}.ply'
         break
       else:
         root = os.path.abspath(f'{root}/../')
+    if mesh_dir == None:
+      raise FileNotFoundError("Can't find lm_models directory, please download it.")
     return mesh_dir