#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导出,用之前得花时间读源码再适配自己的环境。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。