Update modeling.py

modeling.py

@@ -8,6 +8,7 @@ import torch
 from .dino_wrapper2 import DinoWrapper
 from .transformer import TriplaneTransformer
 from .synthesizer_part import TriplaneSynthesizer
+from .processor import LRMImageProcessor
 
 class CameraEmbedder(nn.Module):
     def __init__(self, raw_dim: int, embed_dim: int):
@@ -21,6 +22,16 @@ class CameraEmbedder(nn.Module):
     def forward(self, x):
         return self.mlp(x)
 
+# the processor
+class LRMProcessor:
+    def __init__(self, source_size=512, *args, **kwargs):
+        self.image_processor = LRMImageProcessor(source_size, *args, **kwargs)
+    
+    def process(self, image, batch_size=1):
+        # preprocess the image
+        processed_image, source_camera = self.image_processor(image)
+        return processed_image, source_camera
+        
 class LRMGeneratorConfig(PretrainedConfig):
     model_type = "lrm_generator"