Reference for ultralytics/models/sam/build.py
Note
Full source code for this file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/sam/build.py. Help us fix any issues you see by submitting a Pull Request 🛠️. Thank you 🙏!
ultralytics.models.sam.build.build_sam_vit_h(checkpoint=None)
Build and return a Segment Anything Model (SAM) h-size model.
Source code in ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_sam_vit_l(checkpoint=None)
Build and return a Segment Anything Model (SAM) l-size model.
Source code in ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_sam_vit_b(checkpoint=None)
Build and return a Segment Anything Model (SAM) b-size model.
Source code in ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_mobile_sam(checkpoint=None)
Build and return Mobile Segment Anything Model (Mobile-SAM).
Source code 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)
Builds the selected SAM model architecture.
Source code in ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_sam(ckpt='sam_b.pt')
Build a SAM model specified by ckpt.