From dbc9a0511033e1cdc4762ff1b82b736ea7465e99 Mon Sep 17 00:00:00 2001 From: An Yan Date: Thu, 25 Apr 2024 19:27:21 -0700 Subject: [PATCH] update --- README.md | 4 +++- SoM/demo_som.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 46d58b8..f10f2a6 100644 --- a/README.md +++ b/README.md @@ -111,7 +111,9 @@ After downloading our data (or preparing your own SoM data), train SoM-LLaVA via `bash scripts/v1_5/finetune.sh` -## :snowflake: Notes for using SoM +## :snowflake: Using SoM +**Note:** Our implementaion is improved over the original SoM repo, by removing overlapping regions for each mask (otherwise there will be confilicts/overlaps for tag positions). + * Init virtual envs ```bash diff --git a/SoM/demo_som.py b/SoM/demo_som.py index fb7f895..81a529c 100644 --- a/SoM/demo_som.py +++ b/SoM/demo_som.py @@ -40,7 +40,7 @@ semsam_cfg = "configs/semantic_sam_only_sa-1b_swinL.yaml" seem_cfg = "configs/seem_focall_unicl_lang_v1.yaml" -model_dir = "/home/abc/checkpoints/" +model_dir = "~/checkpoints/" semsam_ckpt = model_dir+"swinl_only_sam_many2many.pth" sam_ckpt = model_dir+"sam_vit_h_4b8939.pth" seem_ckpt = model_dir+"seem_focall_v1.pt"