Riferimento per ultralytics/models/sam/model.py
Nota
Questo file è disponibile su https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ sam/model .py. Se noti un problema, contribuisci a risolverlo inviando una Pull Request 🛠️. Grazie 🙏!
ultralytics.models.sam.model.SAM
Basi: Model
SAM (Segment Anything Model) classe di interfaccia.
SAM è stato progettato per la segmentazione di immagini in tempo reale. Può essere utilizzato con una varietà di suggerimenti, come ad esempio bounding box, punti o etichette. Il modello è in grado di garantire prestazioni a zero scatti ed è stato addestrato sul dataset SA-1B. e viene addestrato sul set di dati SA-1B.
Codice sorgente in ultralytics/models/sam/model.py
task_map
property
Fornisce una mappatura dal compito "segmento" al corrispondente "predittore".
Restituzione:
Tipo | Descrizione |
---|---|
dict
|
Un dizionario che mappa il compito "segmento" con il corrispondente "Predittore". |
__call__(source=None, stream=False, bboxes=None, points=None, labels=None, **kwargs)
Alias per il metodo 'predict'.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
source |
str
|
Percorso del file immagine o video, o un oggetto PIL.Image, o un oggetto numpy.ndarray. |
None
|
stream |
bool
|
Se Vero, attiva lo streaming in tempo reale. L'impostazione predefinita è Falso. |
False
|
bboxes |
list
|
Elenco delle coordinate del rettangolo di selezione per la segmentazione richiesta. Il valore predefinito è Nessuno. |
None
|
points |
list
|
Elenco di punti per la segmentazione richiesta. Il valore predefinito è Nessuno. |
None
|
labels |
list
|
Elenco di etichette per la segmentazione richiesta. Il valore predefinito è Nessuno. |
None
|
Restituzione:
Tipo | Descrizione |
---|---|
list
|
Le previsioni del modello. |
Codice sorgente in ultralytics/models/sam/model.py
__init__(model='sam_b.pt')
Inizializza il modello SAM con un file di modello pre-addestrato.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
model |
str
|
Percorso del file del modello pre-addestrato SAM . Il file deve avere un'estensione .pt o .pth. |
'sam_b.pt'
|
Aumenta:
Tipo | Descrizione |
---|---|
NotImplementedError
|
Se l'estensione del file del modello non è .pt o .pth. |
Codice sorgente in ultralytics/models/sam/model.py
info(detailed=False, verbose=True)
Registra le informazioni sul modello SAM .
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
detailed |
bool
|
Se Vero, visualizza informazioni dettagliate sul modello. L'impostazione predefinita è False. |
False
|
verbose |
bool
|
Se Vero, visualizza le informazioni sulla console. Il valore predefinito è Vero. |
True
|
Restituzione:
Tipo | Descrizione |
---|---|
tuple
|
Una tupla contenente le informazioni del modello. |
Codice sorgente in ultralytics/models/sam/model.py
predict(source, stream=False, bboxes=None, points=None, labels=None, **kwargs)
Esegue la previsione della segmentazione sull'immagine o sulla sorgente video data.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
source |
str
|
Percorso del file immagine o video, o un oggetto PIL.Image, o un oggetto numpy.ndarray. |
richiesto |
stream |
bool
|
Se Vero, attiva lo streaming in tempo reale. L'impostazione predefinita è Falso. |
False
|
bboxes |
list
|
Elenco delle coordinate del rettangolo di selezione per la segmentazione richiesta. Il valore predefinito è Nessuno. |
None
|
points |
list
|
Elenco di punti per la segmentazione richiesta. Il valore predefinito è Nessuno. |
None
|
labels |
list
|
Elenco di etichette per la segmentazione richiesta. Il valore predefinito è Nessuno. |
None
|
Restituzione:
Tipo | Descrizione |
---|---|
list
|
Le previsioni del modello. |