from IPython.display import clear_output
%cd ~
!git clone -b develop https://github.com/PaddlePaddle/PaddleNLP.git
clear_output() # 清理很長的內(nèi)容

%cd ~/PaddleNLP
!pip install -e ./
clear_output() # 清理很長的內(nèi)容
print('安裝完畢, 請(qǐng)重啟內(nèi)核')

安裝完畢, 請(qǐng)重啟內(nèi)核

%cd ~/PaddleNLP/ppdiffusers/
!pip install -r ~/PaddleNLP/ppdiffusers/requirements.txt
clear_output() # 清理很長的內(nèi)容
print('安裝完畢, 請(qǐng)重啟內(nèi)核')

%cd ~/PaddleNLP/ppdiffusers

/home/aistudio/PaddleNLP/ppdiffusers

import paddle
from ppdiffusers import StableDiffusionPipeline
# 加載模型
pipe = StableDiffusionPipeline.from_pretrained("Linaqruf/anything-v3.0")

# prompt： 戴眼鏡坐在電腦邊敲代碼穿唐裝的兔子， 可愛，大眼睛，高清毛絨
prompt = "rabbit wear glasses  beside the computer is writting code  in Tang costume for the Spring Festival is cute, big-eyed, high-definition plush"
images = pipe(prompt, guidance_scale=7.5, height=768, width=768)
print(len(images.images))
image = images.images[0]
image.save("/home/aistudio/0.png")

%cd ~
import cv2
def merge():
    back_img_path = '底圖.png'
    # 加載底圖模板
    back_img = cv2.imread(back_img_path, cv2.IMREAD_UNCHANGED)  # 捕獲圖像1
    print('底圖尺寸: ', back_img.shape)
    # 加載生成圖
    fore_img = cv2.imread('0.png', cv2.IMREAD_UNCHANGED)
    fore_img = cv2.resize(fore_img, (600, 600))
    rows, cols, channels = fore_img.shape
    print('生成圖尺寸: ', fore_img.shape)
    dx, dy = 75, 390
    print('dx + rows,  dy + cols', dx + rows, dy + cols)
    print(back_img[dy:dy + cols, dx:dx + rows].shape)
    # 合并
    back_img[dy:dy + cols, dx:dx + rows] = fore_img
    merge_img_path = '海報(bào).png'
    # 保存
    cv2.imwrite(merge_img_path, back_img)
    return merge_img_path
merge()

/home/aistudio
底圖尺寸:  (1334, 750, 3)
生成圖尺寸:  (600, 600, 3)
dx + rows,  dy + cols 675 990
(600, 600, 3)
'海報(bào).png'

import cv2
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
img = cv2.imread("海報(bào).png", -1)#讀取的是bgr格式
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
plt.imshow(img)
plt.show()