Riferimento per ultralytics/models/sam/build.py
Nota
Questo file è disponibile su https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ sam/build .py. Se noti un problema, contribuisci a risolverlo inviando una Pull Request 🛠️. Grazie 🙏!
ultralytics.models.sam.build.build_sam_vit_h(checkpoint=None)
Costruisce e restituisce un modello Segment Anything Model (SAM) di dimensione h.
Codice sorgente in ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_sam_vit_l(checkpoint=None)
Costruisce e restituisce un modello Segment Anything Model (SAM) di dimensioni l.
Codice sorgente in ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_sam_vit_b(checkpoint=None)
Costruisce e restituisce un modello Segment Anything Model (SAM) di dimensione b.
Codice sorgente in ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_mobile_sam(checkpoint=None)
Costruire e restituire il Mobile Segment Anything Model (Mobile-SAM).
Codice sorgente in ultralytics/models/sam/build.py
ultralytics.models.sam.build._build_sam(encoder_embed_dim, encoder_depth, encoder_num_heads, encoder_global_attn_indexes, checkpoint=None, mobile_sam=False)
Costruisce l'architettura del modello SAM selezionato.
Codice sorgente in ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_sam(ckpt='sam_b.pt')
Costruisci un modello SAM specificato da ckpt.