add model

config.json

@@ -2,6 +2,10 @@
   "architectures": [
     "MonoSceneModel"
   ],
+  "auto_map": {
+    "AutoConfig": "config.MonoSceneConfig",
+    "AutoModel": "monoscene_model.MonoSceneModel"
+  },
   "avg_down": false,
   "base_width": 64,
   "block_type": "bottleneck",

config.py

@@ -0,0 +1,26 @@
+from transformers import PretrainedConfig
+from typing import List
+
+
+class MonoSceneConfig(PretrainedConfig):
+
+    def __init__(
+        self,
+        dataset="kitti",
+        n_classes=20,
+        feature=64,
+        project_scale=2,
+        full_scene_size=(256, 256, 32),
+        **kwargs,
+    ):
+        self.dataset = dataset
+        self.n_classes = n_classes
+        self.feature = feature
+        self.project_scale = project_scale
+        self.full_scene_size = full_scene_size
+        super().__init__(**kwargs)
+
+
+
+
+

monoscene_model.py

@@ -0,0 +1,21 @@
+from transformers import PreTrainedModel
+from .config import MonoSceneConfig
+from monoscene.monoscene import MonoScene
+
+
+class MonoSceneModel(PreTrainedModel):
+    config_class = MonoSceneConfig
+
+    def __init__(self, config):
+        super().__init__(config)
+        self.model = MonoScene(
+            dataset=config.dataset,
+            n_classes=config.n_classes,
+            feature=config.feature,
+            project_scale=config.project_scale,
+            full_scene_size=config.full_scene_size
+        )
+     
+
+    def forward(self, tensor):
+        return self.model.forward(tensor)

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:11d1a04f68fcd0a49028c907393eaeff7fb0d003ffd0b8d8dfdba520cd76b8c7
+oid sha256:5bdc2fc2fb305da11cf851114e7e6784c99ae917fa5265f90e3ff3d47cc8b193
 size 600200087