PaddleHub DeepLabv3+模型人像抠图

深度学习
通过加载PaddleHub DeepLabv3+模型(deeplabv3p_xception65_humanseg)实现一键抠图
  1. import paddlehub as hub
  2. humanseg = hub.Module(name=”deeplabv3p_xception65_humanseg”)
  3. import matplotlib.pyplot as plt
  4. import matplotlib.image as mpimg
  5. path = [‘work/timg.jpg’]
  6. results = humanseg.segmentation(data={“image”:path})
  1. # 原图
  2. img = mpimg.imread(path[0])
  3. # 展示预测结果图片
  4. plt.figure(figsize=(10,10))
  5. plt.imshow(img)
  6. plt.axis(‘off’)
  7. plt.show()

 

  1. # 预测结果展示
  2. print(results)
  3. for i in results:
  4.     test_img_path = i[‘processed’]
  5.     img = mpimg.imread(test_img_path)
  6.     # 展示预测结果图片
  7.     plt.figure(figsize=(10,10))
  8.     plt.imshow(img)
  9.     plt.axis(‘off’)
  10.     plt.show()

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注