commit
Browse files
dust3r/cloud_opt_flow/optimizer.py
CHANGED
|
@@ -11,7 +11,7 @@ from dust3r.utils.geometry import xy_grid, geotrf, depthmap_to_pts3d
|
|
| 11 |
from dust3r.utils.device import to_cpu, to_numpy
|
| 12 |
from dust3r.utils.goem_opt import DepthBasedWarping, OccMask, WarpImage, depth_regularization_si_weighted, tum_to_pose_matrix
|
| 13 |
from third_party.raft import load_RAFT
|
| 14 |
-
from sam2.build_sam import build_sam2_video_predictor
|
| 15 |
sam2_checkpoint = "third_party/sam2/checkpoints/sam2.1_hiera_large.pt"
|
| 16 |
model_cfg = "configs/sam2.1/sam2.1_hiera_l.yaml"
|
| 17 |
|
|
|
|
| 11 |
from dust3r.utils.device import to_cpu, to_numpy
|
| 12 |
from dust3r.utils.goem_opt import DepthBasedWarping, OccMask, WarpImage, depth_regularization_si_weighted, tum_to_pose_matrix
|
| 13 |
from third_party.raft import load_RAFT
|
| 14 |
+
from third_party.sam2.build_sam import build_sam2_video_predictor
|
| 15 |
sam2_checkpoint = "third_party/sam2/checkpoints/sam2.1_hiera_large.pt"
|
| 16 |
model_cfg = "configs/sam2.1/sam2.1_hiera_l.yaml"
|
| 17 |
|
third_party/RAFT/models/Tartan-C-T432x960-M.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:86b45984bd8326f9d9501646884d931977ac9961c8171e5714a52bd6bc78d03d
|
| 3 |
+
size 78894026
|