百度飞桨系列文章：
关注专栏：百度飞桨
春联，又称“春贴”、“门对”、“对联”，是过年时所贴的红色喜庆元素“年红”中一个种类。它以对仗工整、简洁精巧的文字描绘美好形象，抒发美好愿望，是中国特有的文学形式，是华人们过年的重要习俗。当人们在自己的家门口贴年红（春联、福字、窗花等）的时候，意味着过春节正式拉开序幕。
每逢春节，无论城市还是农村，家家户户都要挑漂亮的红春联贴于门上，辞旧迎新，增加喜庆的节日气氛。春联的另一来源是春贴，古人在立春日多贴“宜春”二字，后渐渐发展为春联，表达了中国劳动人民一种辟邪除灾、迎祥纳福的美好愿望。
本项目基于PaddleHub的对联生成模型ernie_tiny_couplet，使用的是paddlepaddle 2.x 框架，根据上联生成对联的下联。
ernie_gen_couplet框架：
ernie_tiny_couplet模型：
代码基于飞桨的 BML CodeLab 编写
# 安装环境依赖
!pip install paddlehub --upgrade -i https://mirror.baidu.com/pypi/simple
import paddlehub as hub
import re
import io
from PIL import Image
import numpy as np
import requests
module = hub.Module(name="ernie_gen_couplet") # 加载模型
# ***************春 联 上 联***************
test_texts_a = ['万事平安幸福年'] #可以通过修改 test_texts_a 中的内容，生成你想要的春联。
test_texts_bs = module.generate(texts=test_texts_a, use_gpu=True, beam_width=1)
test_texts_b = test_texts_bs[0]
# 获取汉字图片
def get_word(ch, quality):
"""
ch - 单个汉字或英文字母（仅支持大写）
quality - 单字分辨率，H-640像素，M-480像素，L-320像素
"""
fp = io.BytesIO(requests.post(url='http://xufive.sdysit.com/tk', data={'ch':ch}).content) # 只包含大约1500常用汉字
im = Image.open(fp)
w, h = im.size
if quality == 'M':
w, h = int(w*0.75), int(0.75*h)
elif quality == 'L':
w, h = int(w*0.5), int(0.5*h)
return im.resize((w,h))
# 生成春联背景图
def get_bg(quality):
return get_word('bg', quality)
# 生成春联
def write_couplets(text, HorV='V', quality='L', out_file=None):
"""
text - 春联内容，以空格断行
HorV - H-横排，V-竖排
quality - 单字分辨率，H-640像素，M-480像素，L-320像素
out_file - 输出文件名
"""
usize = {'H':(640,23), 'M':(480,18), 'L':(320,12)}
bg_im = get_bg(quality)
text_list = [list(item) for item in text.split()]
rows = len(text_list)
cols = max([len(item) for item in text_list])
if HorV == 'V':
ow, oh = 40+rows*usize[quality][0]+(rows-1)*10, 40+cols*usize[quality][0]
else:
ow, oh = 40+cols*usize[quality][0], 40+rows*usize[quality][0]+(rows-1)*10
out_im = Image.new('RGBA', (ow, oh), '#f0f0f0')
for row in range(rows):
if HorV == 'V':
row_im = Image.new('RGBA', (usize[quality][0], cols*usize[quality][0]), 'white')
offset = (ow-(usize[quality][0]+10)*(row+1)-10, 20)
else:
row_im = Image.new('RGBA', (cols*usize[quality][0], usize[quality][0]), 'white')
offset = (20, 20+(usize[quality][0]+10)*row)
for col, ch in enumerate(text_list[row]):
if HorV == 'V':
pos = (0, col*usize[quality][0])
else:
pos = (col*usize[quality][0],0)
ch_im = get_word(ch, quality)
row_im.paste(bg_im, pos)
row_im.paste(ch_im, (pos[0]+usize[quality][1], pos[1]+usize[quality][1]), mask=ch_im)
out_im.paste(row_im, offset)
if out_file:
out_im.convert('RGB').save(out_file)
out_im.show()
# 开始写春联
text = test_texts_b,test_texts_a
text=re.sub('[^\u4e00-\u9fa5]+',' ',str(text))
write_couplets(text, HorV='V', quality='M', out_file='春联.jpg')
