基于SAM的one-shot提取特定目标

发布时间 2023-07-14 14:10:07作者: 海_纳百川

项目地址:https://github.com/ZrrSkywalker/Personalize-SAM

实现功能:仅给定一张带有参考mask的图像,PerSAM 无需任何训练即可在其他图像或视频中分割特定的目标,例如您的宠物狗。也提供了微调,但是只训练了2个参数

论文中图片介绍:

根据demo地址测试(https://huggingface.co/spaces/justin-zk/Personalize-SAM):

官方提供的图片测试

 

自己提供的图片测试:第一张测试结果不好,没有分割出布达拉宫,第二张测试结果准确。