#359 · 主分类: 计算机视觉

finetune-anything

computer-vision deep-learning fine-tune segment-anything

微调SAM(Segment Anything Model)以用于特定场景下的计算机视觉任务,如语义分割、抠图、检测等。

项目最后更新:08/05/23

GitHub Stars

868

Forks数量

62

贡献者数量

2

许可证

MIT

收录理由

想在自己的数据上微调SAM(Segment Anything Model)做特定场景的分割,这个项目能把大部分流程自动化。它把SAM的图像编码器、提示编码器和掩码解码器都封装成可定制的适配器,你可以选择冻结部分参数,只用配置文件加一个任务名就能跑训练和评估。目前语义分割这条路径已经能用了,支持torchvision风格的数据集做训练和验证;抠图、实例分割和检测还在计划中。如果你的团队想复用SAM而不是从头搭流水线,这是个不错的起点。不过要留意,代码还处于早期研究阶段,维护者少,也还没支持ONNX导出,用之前得花时间读源码再适配自己的环境。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目