EfficientSAM 代码推理

SA网站主页:Segment Anything | Meta AI

论文主页:EfficientSAM

代码地址:https://github.com/yformer/EfficientSAM

官方给的推理代码是CPU版本的,如果想使用GPU推理需要自己修改一下

EfficientSAM 代码推理_第1张图片

EfficientSAM 代码推理_第2张图片

经过推理测试3090GPU,官方测试图片1072*603 使用框或者点prompts 使用vitt模型需要41-58ms左右,显存7G左右

使用全图分割耗时会更长

EfficientSAM 代码推理_第3张图片

你可能感兴趣的:(人工智能,分割)