在前面的文章中使用了rembg,一个非常好用且开源的图像分割工具:使用rembg并捣鼓ONNX Runtime - cyqsd's blog,也提到了birefnet、u2net_xxx等适用于各类场景的模型,但毕竟那都是人家预训练的模型,并不能涵盖现实使用时的所有场景,并且要达到较好效果的预训练模型,往往体积也很大,不利于轻量化的使用。那这时可以考虑自己构建训练集,在issues的提问中有人问到

- 阅读全文 -