code: def __init__(self):
      super(U2Net_Portrait, self).__init__()
      self.model = U2NET(3, 1)
      state_dict = paddle.load(
          os.path.join(self.directory, "u2net_portrait.pdparams")
      )
      self.model.set_dict(state_dict)
      self.model.eval()
  
file_path: PaddleHub/modules/image/Image_gan/style_transfer/U2Net_Portrait/module.py
imports: import os
  import paddle
  import paddle.nn
  import numpy
  from U2Net_Portrait.u2net import U2NET
  from U2Net_Portrait.processor import Processor
  from paddlehub.module.module import moduleinfo
label_desc: Use image node_id 1 to create a new U2Net_Portrait object from inherited base classes: nn with example: obj = U2Net_Portrait()