Pytorch-U2net 你可以用这个图像分割模型做人像背景分割,智能抠图等。 基于Pytorch与U-net,我写了两个模型,一个是大模型U2net,效果比较好。一个是小模型U2NETP ,效果稍差。 Train 数据集获取: 链接:https://pan.baidu.com/s/1wFxSswxDSY9VS3FBKyTPxw 提取码:sofm Inference 权重获取: 链接:https://pan.baidu.com/s/1vE-0kYD3oddqlMdVo1FMAg 提取码:sofm