prototype pipeline

models/depth_estim.py

@@ -0,0 +1,17 @@
+import os
+import sys
+
+import torch
+import torch.nn as nn
+
+
+class DepthEstimation(nn.Module):
+    def __init__(self):
+        super().__init__()
+
+        self.layer_1 = nn.Linear(in_features=3, out_features=3)  
+
+    def forward(self, x: torch.Tensor) -> torch.Tensor:
+        x = self.layer_1(x)
+        
+        return x