From acaf210abbbdca1d897bf1f07931fdb100abe55c Mon Sep 17 00:00:00 2001 From: chflame163 Date: Tue, 3 Sep 2024 14:43:30 +0800 Subject: [PATCH] remove SAM2 Video Ultra debug info --- py/sam_2_ultrl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py/sam_2_ultrl.py b/py/sam_2_ultrl.py index 4a784cd..3635d5b 100644 --- a/py/sam_2_ultrl.py +++ b/py/sam_2_ultrl.py @@ -477,7 +477,7 @@ class LS_SAM2_VIDEO_ULTRA: process_detail, device, max_megapixels, pre_mask=None ): - debug_image = "e:\\tmp\\debug.jpg" + if len(bboxes) == 0: log(f"{self.NODE_NAME} skipped, because bboxes is empty.", message_type='error') @@ -576,7 +576,7 @@ class LS_SAM2_VIDEO_ULTRA: out_list.append(mask_tensor) mask_tensor = torch.stack(out_list, dim=0).cpu().float() first_frame_mask = tensor2pil(mask_tensor.squeeze()).convert("L") - first_frame_mask.save(debug_image) + coords = poisson_disk_sampling(first_frame_mask, radius=32, num_points=16) log(f"coords = {coords}")