Update processor.py

processor.py

@@ -19,9 +19,7 @@ import kiui
 
 class LRMImageProcessor(BaseImageProcessorFast):
     def __init__(self, source_size=512, *args, **kwargs):
-        print("add super")
         super().__init__(*args, **kwargs)
-        print("Add source")
         self.source_size = source_size
         self.session = None
         self.rembg_remove = None
@@ -54,7 +52,6 @@ class LRMImageProcessor(BaseImageProcessorFast):
         return fx, fy, cx, cy
 
     def build_camera_principle(self, RT: torch.Tensor, intrinsics: torch.Tensor):
-        print("build_camera_principle")
         fx, fy, cx, cy = self.get_normalized_camera_intrinsics(intrinsics)
         return torch.cat([
             RT.reshape(-1, 12),
@@ -65,7 +62,6 @@ class LRMImageProcessor(BaseImageProcessorFast):
         ], dim=-1)
 
     def _default_intrinsics(self):
-        print("_default_intrinsics")
         fx = fy = 384
         cx = cy = 256
         w = h = 512
@@ -77,7 +73,6 @@ class LRMImageProcessor(BaseImageProcessorFast):
         return intrinsics
 
     def _default_source_camera(self, batch_size: int = 1):
-        print("_default_source_camera")
         dist_to_center = 1.5
         canonical_camera_extrinsics = torch.tensor([[
             [0, 0, 1, 1],
@@ -89,8 +84,6 @@ class LRMImageProcessor(BaseImageProcessorFast):
         return source_camera.repeat(batch_size, 1)
 
     def __call__(self, image, *args, **kwargs):
-        print("processing...")
         processed_image = self.preprocess_image(image)
-        print("processed done")
         source_camera = self._default_source_camera(batch_size=1)
         return processed_image, source_camera