fix bug

app.py

@@ -34,7 +34,8 @@ GRADIO_OBJ_SHADING_PATH = 'gradio_output_shading.obj'
 
 #opt = tyro.cli(AllConfigs)
 
-ckpt_path = hf_hub_download(repo_id="rgxie/LDM", filename="LDM6v01.ckpt")
+#ckpt_path = hf_hub_download(repo_id="rgxie/LDM", filename="LDM6v01.ckpt")
+ckpt_path = '/ssd3/xrg/tensor23d/pretrained/last_6view_0610_14.ckpt'
 
 opt = Options(
     input_size=512, 

core/tensorBase.py

@@ -205,8 +205,23 @@ class TensorBase(torch.nn.Module):
         self.near_far = near_far
         self.step_ratio = 0.9 #step_ratio原作0.5
 
+        print("aabb", self.aabb.view(-1))
+        print("grid size", gridSize)
+        self.aabbSize = self.aabb[1] - self.aabb[0]
+        print(self.aabbSize.device)
+        self.invaabbSize = 2.0/self.aabbSize
+        print(self.invaabbSize.device)
+        # self.invaabbSize = self.invaabbSize.to(self.aabb.device)
+        # print(self.invaabbSize.device)
+        self.gridSize= gridSize.float()
+        self.units=self.aabbSize / (self.gridSize-1)
+        self.stepSize=torch.mean(self.units)*self.step_ratio  # TBD step_ratio? why so small 0.5
+        self.aabbDiag = torch.sqrt(torch.sum(torch.square(self.aabbSize)))
+        self.nSamples=int((self.aabbDiag / self.stepSize).item()) + 1
+        print("sampling step size: ", self.stepSize)
+        print("sampling number: ", self.nSamples)
 
-        self.update_stepSize(gridSize)
+        # self.update_stepSize(gridSize)
 
         self.matMode = [[0,1], [0,2], [1,2]]
         self.vecMode =  [2, 1, 0]