wget images used for sample

NVIDIA · Nov 27, 2024 · 390625e · 390625e
1 parent e2690ce
commit 390625e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/tripy/examples/segment-anything-model-v2/README.md b/tripy/examples/segment-anything-model-v2/README.md
@@ -12,6 +12,7 @@ This is an implementation of SAM2 model ([original repository](https://github.co
 
     ```bash
     sudo apt-get update && sudo apt-get install ffmpeg libsm6 libxext6  -y
+    wget -O truck.jpg https://raw.githubusercontent.com/facebookresearch/sam2/main/notebooks/images/truck.jpg
     mkdir checkpoints && cd checkpoints && wget https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_large.pt
     python3 -m pip install -r requirements.txt
     ```

diff --git a/tripy/examples/segment-anything-model-v2/truck.jpg b/tripy/examples/segment-anything-model-v2/truck.jpg
diff --git a/tripy/tests/test_examples.py b/tripy/tests/test_examples.py
@@ -80,7 +80,10 @@ def __str__(self):
         return os.path.relpath(self.path, EXAMPLES_ROOT)
 
 
-EXAMPLES = [Example(["nanogpt"]), Example(["segment-anything-model-v2"])]
+EXAMPLES = [
+    Example(["nanogpt"]),
+    Example(["segment-anything-model-v2"], artifact_names=["truck.jpg", "saved_engines/", "output/"]),
+]
 
 
 @pytest.mark.l1