Tangledup-ai/sam3_local
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README with arXiv bibtex

Browse Source 
Reviewed By: jayleicn

Differential Revision:
D87813153

Privacy Context Container: L1256182

fbshipit-source-id: 9361ff55ebdb1ee78f694cb9c41b8bc83bf600fb
This commit is contained in:
Haitham Khedr
2025-11-24 13:30:21 -08:00
committed by meta-codesync[bot]
parent d0b1b9d5aa
commit 1daff5eb92
3 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
sam3/model/utils/sam2_utils.py
View File

@@ -97,8 +97,8 @@ def load_video_frames(
video_path,
image_size,
offload_video_to_cpu,
img_mean=(0.485, 0.456, 0.406),
img_std=(0.229, 0.224, 0.225),
img_mean=(0.5, 0.5, 0.5),
img_std=(0.5, 0.5, 0.5),
async_loading_frames=False,
compute_device=torch.device("cuda"),
):
@@ -138,8 +138,8 @@ def load_video_frames_from_jpg_images(
video_path,
image_size,
offload_video_to_cpu,
img_mean=(0.485, 0.456, 0.406),
img_std=(0.229, 0.224, 0.225),
img_mean=(0.5, 0.5, 0.5),
img_std=(0.5, 0.5, 0.5),
async_loading_frames=False,
compute_device=torch.device("cuda"),
):
@@ -205,8 +205,8 @@ def load_video_frames_from_video_file(
video_path,
image_size,
offload_video_to_cpu,
img_mean=(0.485, 0.456, 0.406),
img_std=(0.229, 0.224, 0.225),
img_mean=(0.5, 0.5, 0.5),
img_std=(0.5, 0.5, 0.5),
compute_device=torch.device("cuda"),
):
"""Load the video frames from a video file."""

4
sam3/visualization_utils.py
View File

@@ -43,8 +43,8 @@ COLORS = generate_colors(n_colors=128, n_samples=5000)
def show_img_tensor(img_batch, vis_img_idx=0):
MEAN_IMG = np.array([0.485, 0.456, 0.406])
STD_IMG = np.array([0.229, 0.224, 0.225])
MEAN_IMG = np.array([0.5, 0.5, 0.5])
STD_IMG = np.array([0.5, 0.5, 0.5])
im_tensor = img_batch[vis_img_idx].detach().cpu()
assert im_tensor.dim() == 3
im_tensor = im_tensor.numpy().transpose((1, 2, 0))