[fix] fix vis_blob bugs

examples/blobctrl/blobctrl_app.py

@@ -63,8 +63,6 @@ if not os.path.exists(Dino_path):
     )
 print(f"Dino checkpoints downloaded to {Dino_path}")
 
-
-
 # download SAM model
 SAM_path = "examples/blobctrl/models/sam/sam_vit_h_4b8939.pth"
 if not os.path.exists(SAM_path):
@@ -173,13 +171,12 @@ head= r"""
 descriptions = r"""
 Official Gradio Demo for <a href=''><b>BlobCtrl: A Unified and Flexible Framework for Element-level Image Generation and Editing</b></a><br>
 🦉 BlobCtrl enables precise, user-friendly element-level visual manipulation. <br>
-Main Features: Element-level Add/Remove/Move/Replace/Enlarge/Shrink.
 """
 
 
 citation = r"""
 If BlobCtrl is helpful, please help to ⭐ the <a href='https://github.com/TencentARC/BlobCtrl' target='_blank'>Github Repo</a>. Thanks!
-[![GitHub Stars](https://img.shields.io/github/stars/TencentARC/BlobCtrl?style=social)](https://github.com/TencentARC/BlobCtrl)
+[![GitHub Stars](https://img.shields.io/github/stars/TencentARC/BlobCtrl?style=social)]()
 ---
 📝 **Citation**
 <br>
@@ -187,9 +184,9 @@ If our work is useful for your research, please consider citing:
 ```bibtex
 @misc{li2025blobctrl,
   title={BlobCtrl: A Unified and Flexible Framework for Element-level Image Generation and Editing}, 
-  author={Yaowei Li, Lingen Li, Zhaoyang Zhang, Xiaoyu Li, Guangzhi Wang, Hongxiang Li, Xiaodong Cun, Ying Shan, Yuexian Zou},
+  author={Yaowei Li},
   year={2025},
-  eprint={2503.13434},
+  eprint={2502.09477},
   archivePrefix={arXiv},
   primaryClass={cs.CV}
 }
@@ -1771,7 +1768,7 @@ with block as demo:
 
     remove_blob_box.change(
         remove_blob_box_func,
-        [editable_blob, original_image, tracking_points, ellipse_lists, ori_result_gallery],
+        [editable_blob, original_image, tracking_points, ellipse_lists, ori_result_gallery, remove_blob_box],
         [editable_blob, ellipse_lists, edited_result_gallery, ori_result_gallery, resize_blob_slider_maintain_aspect_ratio]
     )
 
@@ -1804,7 +1801,7 @@ with block as demo:
     ## upload user-specified object image
     init_object_image.upload(
         upload_object_image,
-        [init_object_image, edited_result_gallery],
+        [init_object_image, edited_result_gallery, remove_blob_box],
         [object_image_gallery, remove_blob_box]
     )
 

examples/blobctrl/utils/utils.py

@@ -211,7 +211,7 @@ def splat_features(
                 scores_viz = d_scores
                 
             n_gaussians = xs.shape[-1]
-            ret.update(visualize_features(xs, ys, viz_size, n_gaussians, scores_viz, **kwargs))
+            ret.update(visualize_features(viz_size, n_gaussians, scores_viz, kwargs.get("viz_colors", None)))
             # from PIL import Image
             # import numpy as np
             # viz_colors = torch.load("BED_CONF_COLORS.pt")
@@ -243,7 +243,7 @@ def splat_features(
 
 @torch.no_grad()
 def visualize_features(viz_size=64, n_gaussians=None, scores=None,
-                        viz_colors=None, **kwargs) -> Dict[str, Tensor]:
+                        viz_colors=None) -> Dict[str, Tensor]:
     n_gaussians = n_gaussians+1
    
     rand_colors = viz_colors is None