Niv

.gitignore

@@ -1,13 +1,19 @@
 __pycache__
 *.vscode
-cfg.yaml
+cfg.yam
 *.pth
 *.pkl
 *.pyc
 *.so
 *.egg-info
+g.txt
 # Ignore weight files but keep the folder structure
 weights/*
-!weights/.gitkeep
 output/
-.claude/
+.claude/
+*.ply
+data/mono/*
+data/pattern_cube/*
+reports/*
+
+*.bak

Utils.py

@@ -1,4 +1,4 @@
-import os, sys, torch, imageio, logging, importlib, argparse
+import os, sys, torch, logging, importlib #, argparse, imageio
 import cv2
 import numpy as np
 import yaml

core/foundation_stereo.py

@@ -1,4 +1,4 @@
-import torch,pdb,logging,timm
+import torch #,pdb,logging,timm
 import torch.nn as nn
 import torch.nn.functional as F
 import numpy as np

core/submodule.py

@@ -619,7 +619,8 @@ def __init__(self, in_planes, ratio=16):
 
     def forward(self, x):
         avg_out = self.fc(self.avg_pool(x))
-        max_out = self.fc(self.max_pool(x))
+        #max_out = self.fc(torch.amax(x, dim=(-2, -1), keepdim=True))
+        max_out = self.fc(self.max_pool(x)) # niv
         out = avg_out + max_out
         return self.sigmoid(out)