参考资料 ultralytics/models/sam/build.py
备注
该文件可在https://github.com/ultralytics/ultralytics/blob/main/ ultralytics/models/ sam/build .py。如果您发现问题,请通过提交 Pull Request🛠️ 帮助修复。谢谢🙏!
ultralytics.models.sam.build.build_sam_vit_h(checkpoint=None)
建立并返回一个 Segment Anything Model (SAM) h 尺寸模型。
源代码 ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_sam_vit_l(checkpoint=None)
建立并返回一个 l 尺寸的 Segment Anything Model (SAM) 模型。
源代码 ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_sam_vit_b(checkpoint=None)
建立并返回一个 Segment Anything Model (SAM) b-size 模型。
源代码 ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_mobile_sam(checkpoint=None)
建立并返回移动细分市场模型(Mobile-SAM )。
源代码 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)
构建选定的SAM 模型架构。
源代码 ultralytics/models/sam/build.py
ultralytics.models.sam.build.build_sam(ckpt='sam_b.pt')
建立一个由 ckpt 指定的SAM 模型。