Fix SAM Predictor example in docs (#21700)

This commit is contained in:
Jing Qiu 2025-08-15 17:05:58 +08:00 committed by GitHub
parent 3f443f2745
commit 23d79250e7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions

View File

@ -123,7 +123,7 @@ The Segment Anything Model can be employed for a multitude of downstream tasks t
results = predictor(points=[900, 370], labels=[1])
# Run inference with multiple points prompt
results = predictor(points=[[400, 370], [900, 370]], labels=[[1, 1]])
results = predictor(points=[[400, 370], [900, 370]], labels=[1, 1])
# Run inference with negative points prompt
results = predictor(points=[[[400, 370], [900, 370]]], labels=[[1, 0]])

View File

@ -543,7 +543,7 @@ class Predictor(BasePredictor):
- The extracted features are stored in the `self.features` attribute for later use.
"""
if self.model is None:
self.setup_model(model=None)
self.setup_model()
self.setup_source(image)
assert len(self.dataset) == 1, "`set_image` only supports setting one image!"
for batch in self.dataset: