mirror of
https://github.com/ultralytics/ultralytics.git
synced 2025-09-15 15:48:41 +08:00
Fix SAM Predictor example in docs (#21700)
This commit is contained in:
parent
3f443f2745
commit
23d79250e7
@ -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]])
|
||||
|
||||
@ -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:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user