怎么得到trimap？

[galabala]

您好，我看教程u2net的文章中没有出现分割成trimap的教程，pymatting不是需要原图和trimap图嘛？

[itainf]

你好，这些代码是不是你需要的？
参考地址：https://zhuanlan.zhihu.com/p/158236640

org_img = "..\\aiphoto\\img\\meinv.jpg"
alpha_img = "..\\aiphoto\\img\\meinv_alpha.png"
alpha_resize_img = "..\\aiphoto\\img\\meinv_alpha_resize.png"

# 通过u_2_net 获取 alpha
my_u2net_test.test_seg_trimap(org_img, alpha_img, alpha_resize_img)

# 通过alpha 获取 trimap
trimap = "..\\aiphoto\\img\\meinv_trimap_resize.png"
to_standard_trimap.to_standard_trimap(alpha_resize_img, trimap)

​
# 证件照添加纯色背景
id_image = "..\aiphoto\img\meinv_id_grid.png"
to_background.to_background_grid(org_img, trimap, id_image)