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

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))