From a16b0760f617cd3da27ae56aa78857148f1250b7 Mon Sep 17 00:00:00 2001 From: Jianyuan Wang Date: Sat, 22 Mar 2025 18:09:30 +0000 Subject: [PATCH] add instruction about masking --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 6dbeba4..6997b27 100644 --- a/README.md +++ b/README.md @@ -109,7 +109,8 @@ with torch.no_grad(): track_list, vis_score, conf_score = model.track_head(aggregated_tokens_list, images, ps_idx, query_points=query_points[None]) ``` -Furthermore, if certain pixels in the input frames are unwanted (e.g., sea, glass), you can mask them by setting the corresponding pixel values to 0 or 1, and then pass the masked images to the model. + +Furthermore, if certain pixels in the input frames are unwanted (e.g., reflective surfaces, sky, or water), you can simply mask them by setting the corresponding pixel values to 0 or 1. Precise segmentation masks aren't necessary - simple bounding box masks work effectively (check this [issue](https://github.com/facebookresearch/vggt/issues/47) for an example). ## Visualization