diff --git a/tripy/examples/segment-anything-model-v2/README.md b/tripy/examples/segment-anything-model-v2/README.md index 2116b29c..b8001f16 100644 --- 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 deleted file mode 100644 index 6b98688c..00000000 Binary files a/tripy/examples/segment-anything-model-v2/truck.jpg and /dev/null differ diff --git a/tripy/tests/test_examples.py b/tripy/tests/test_examples.py index 8f216cd0..e9f9b0a5 100644 --- 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