init project

modules/dust3r/cloud_opt/commons.py

@@ -49,19 +49,42 @@ def get_imshapes(edges, pred_i, pred_j):
     return imshapes
 
 
+# def get_conf_trf(mode):
+#     if mode == 'log':
+#         def conf_trf(x): return x.log()
+#     elif mode == 'sqrt':
+#         def conf_trf(x): return x.sqrt()
+#     elif mode == 'm1':
+#         def conf_trf(x): return x-1
+#     elif mode in ('id', 'none'):
+#         def conf_trf(x): return x
+#     else:
+#         raise ValueError(f'bad mode for {mode=}')
+#     return conf_trf
+
+def conf_trf_log(x):
+    return x.log()
+
+def conf_trf_sqrt(x):
+    return x.sqrt()
+
+def conf_trf_m1(x):
+    return x - 1
+
+def conf_trf_id(x):
+    return x
+
 def get_conf_trf(mode):
     if mode == 'log':
-        def conf_trf(x): return x.log()
+        return conf_trf_log
     elif mode == 'sqrt':
-        def conf_trf(x): return x.sqrt()
+        return conf_trf_sqrt
     elif mode == 'm1':
-        def conf_trf(x): return x-1
+        return conf_trf_m1
     elif mode in ('id', 'none'):
-        def conf_trf(x): return x
+        return conf_trf_id
     else:
-        raise ValueError(f'bad mode for {mode=}')
-    return conf_trf
-
+        raise ValueError(f"bad mode {mode=}")
 
 def l2_dist(a, b, weight=None):
     if weight == None: