20201223:パドルパドルカットアウトモデルを呼び出して、写真の背景色を変更します

1.環境への設置

  • pip install paddlepaddle -i https://pypi.tuna.tsinghua.edu.cn/simple
    (注:CPU環境にはインストールできません。GPU環境にインストールする必要があります)
  • pip install paddlehub

2.スクリプトは次のとおりです


import matplotlib.pyplot as plt 
import matplotlib.image as mpimg

photo_path = ['mines.jpg']

#显示原图
img = mpimg.imread(photo_path[0])
plt.figure(figsize=(10,10))
plt.imshow(img) 
plt.axis('off') 
plt.show()


#改变底色
import paddlehub as hub

output_path = 'output'
module = hub.Module(name="deeplabv3p_xception65_humanseg")
input_dict = {"image": photo_path}
results = module.segmentation(data=input_dict,visualization=True,
                 output_dir=output_path)



#显示处理好的图片
import glob
import os
img_path = glob.glob(os.path.join(output_path, '*.png'))
img_path.sort()

for path in img_path:
    img= mpimg.imread(path)

plt.figure(figsize=(10,10))
plt.imshow(img) 
plt.axis('off') 
plt.show()

おすすめ

転載: blog.csdn.net/weixin_38192254/article/details/111595283