跳到内容

晚上好,辛苦一天了,放松一下吧。

  • 4 回复
  • 457 浏览

手残党也能跑的AI抠图项目

参与讨论

刚试了briaai的RMBG v1.4,开源模型把图片背景抠得贼干净。部署就一个pip install加几行Python代码,MacBook都能跑。GitHub搜RMBG-1.4就能找到,别下错那个2.0的商用版。我测试时发现输入图片别太大,不然内存直接炸。

讨论区

按楼层回复,支持引用与表情;使用 @用户名 或 @昵称 可提醒对方;发言计入圈子贡献字数。

4s 条评论

  1. 粉笔没落灰
    粉笔没落灰 成长8,027

    @甲方爸爸醒醒 我试了试,pip install transformers torch就行,但记得把图片resize到1024以内,不然内存直接炸,班上那台老Mac差点翻车。

  2. 安全佬路过
    安全佬路过 成长4,106
    引用 粉笔没落灰

    @粉笔没落灰 resize到1024确实稳,我试了直接输原图4K直接OOM。另外补充一下,RMBG v1.4的权重文件大概176MB,git lfs拉的话记得先装lfs,不然下个假文件白跑。

  3. 路过点个赞
    路过点个赞 成长5,256
    引用 粉笔没落灰

    我上次也是在这踩坑了,直接pip install briaai/RMBG-1.4那个模型库,但权重得手动下,不然跑出来的图全是灰的。有人用onnx版本跑过没,听说能省点内存。

  4. 一句封神
    一句封神 成长7,037
    引用 路过点个赞

    我上次也是在这踩坑了,直接pip install briaai/RMBG-1.4那个模型库,但权重得手动下,不然跑出来的图全是灰的。有人用onnx版…

    @路过点个赞 onnx版本我试过,内存占用确实低一截,但得装onnxruntime和onnx,模型导出一行代码就搞定,GitHub上那个example跑得挺顺。

发表评论

登录后发表评论

登录即可参与楼层讨论,支持引用回复与 @ 提醒。

前往登录页

Welcome! This site is in Chinese. Tap EN in the top bar to read in English.