$ hub install U2Netp==1.0.0
def Segmentation(
images=None,
paths=None,
batch_size=1,
input_size=320,
output_dir='output',
visualization=False):
图像前景背景分割 API
参数
返回
代码示例
import cv2
import paddlehub as hub
model = hub.Module(name='U2Netp')
result = model.Segmentation(
images=[cv2.imread('/PATH/TO/IMAGE')],
paths=None,
batch_size=1,
input_size=320,
output_dir='output',
visualization=True)
https://github.com/NathanUA/U-2-Net
paddlepaddle >= 2.0.0rc0
paddlehub >= 2.0.0b1