From c41f211309f0605a6b47c9f0b6fc759e00157267 Mon Sep 17 00:00:00 2001
From: xiaoyuxi <xiaoyuxi@bytedance.com>
Date: Tue, 8 Jul 2025 16:06:29 +0800
Subject: [PATCH] first-commit

---
 README.md                                          | 2 +-
 inference.py                                       | 8 --------
 models/SpaTrackV2/models/tracker3D/TrackRefiner.py | 1 -
 3 files changed, 1 insertion(+), 10 deletions(-)

diff --git a/README.md b/README.md
index 2a9798e..635e7b3 100644
--- a/README.md
+++ b/README.md
@@ -71,7 +71,7 @@ By following these steps, you should have a working environment ready to run the
 We gave two examples to illustrate the usage of `SpaTrack2`. 
 ### Type1: Monocular video as input *(Example0)*
 ```
-python inference.py --data_type="RGB" --data_dir="examples" --video_name="protein" --fps=3
+python inference.py --data_type="RGB" --data_dir="examples" --video_name="protein" --fps=5
 ```
 
 ### Type2: Customized Posed RGBD video as input *(Example1)*
diff --git a/inference.py b/inference.py
index bfd7b52..4c98e15 100644
--- a/inference.py
+++ b/inference.py
@@ -170,14 +170,6 @@ if __name__ == "__main__":
                     depth_tensor = T.Resize((new_h, new_w))(torch.from_numpy(depth_tensor))
 
         if viz:
-            for i in range(c2w_traj.shape[0]):
-                img = (video)[i].permute(1,2,0).cpu().numpy()
-                img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-                cv2.imwrite(os.path.join(out_dir, f'frame_{i:04d}.png'), img)
-                point_map_i = point_map[i].cpu().numpy()
-                np.save(os.path.join(out_dir, f'point_{i:04d}.npy'), point_map_i)
-                np.save(os.path.join(out_dir, f'frame_{i:04d}.npy'), point_map_i[0])
-            
             viser.visualize(video=video[None],
                                 tracks=track2d_pred[None][...,:2],
                                 visibility=vis_pred[None],filename="test")
diff --git a/models/SpaTrackV2/models/tracker3D/TrackRefiner.py b/models/SpaTrackV2/models/tracker3D/TrackRefiner.py
index 251e864..cb1d5f7 100644
--- a/models/SpaTrackV2/models/tracker3D/TrackRefiner.py
+++ b/models/SpaTrackV2/models/tracker3D/TrackRefiner.py
@@ -24,7 +24,6 @@ from models.SpaTrackV2.models.tracker3D.spatrack_modules.utils import (
 )
 from models.SpaTrackV2.models.tracker3D.spatrack_modules.ba import extract_static_from_3DTracks, ba_pycolmap
 from models.SpaTrackV2.models.tracker3D.spatrack_modules.pointmap_updator import PointMapUpdator
-from models.SpaTrackV2.models.depth_refiner.depth_refiner import TrackStablizer
 from models.SpaTrackV2.models.tracker3D.spatrack_modules.alignment import affine_invariant_global_loss
 from models.SpaTrackV2.models.tracker3D.delta_utils.upsample_transformer import UpsampleTransformerAlibi