[AI创造营]Wechaty实用小工具---证件照助手

# 0.8.15版本不太好使 切勿踩坑
!pip install wechaty==0.8.11

# 证件照生成函数
def makeZjz(img):
    pic_path = img
    face_landmark = hub.Module(name="face_landmark_localization")
    human_seg = hub.Module(name="deeplabv3p_xception65_humanseg")
    rate = 1.3
    thresh = 2
    # 人脸识别
    result = face_landmark.keypoint_detection(paths=[pic_path])
    face = np.array(result[0]['data'][0], dtype=np.int64)
    # 剪裁
    left = face[:, 0].min()
    right = face[:, 0].max()
    w = right - left
    cw = int((right + left) / 2)
    upper = face[:, 1].min()
    lower = face[:, 1].max()
    h = lower - upper
    ch = int((lower + upper) / 2)
    h = int(413 * w / 295)
    box = (cw - rate * w, ch - rate * h, cw + rate * w, ch + rate * h)
    img = Image.open(pic_path)
    img = img.crop(box)
    img = img.resize((295, 413), Image.ANTIALIAS)
    result = human_seg.segmentation(images=[cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)],
                                    use_gpu=False,
                                    visualization=True,
                                    output_dir='humanseg_output')
    path = result[0]["save_path"]
    print("humanseg path:",path)
    # 上色
    for pic in ["red.png","blue.png","white.png"]:
        # 获取图片,方便后面的代码调用
        frame = Image.open(pic)
        # 给图片指定色彩显示格式
        logo = Image.open(path)
        logo = logo.convert("RGBA")
        frame.paste(logo, (0, 20), mask=logo)
        # 保存图片
        frame.save("result/final_"+pic)
        print("结果：result/final_"+pic)

# Paddlehub文本模型
text_model1 = hub.Module(name='ernie_gen_lover_words')     # 情话模型
text_model2 = hub.Module(name="ernie_gen_acrostic_poetry", line=4, word=7)   # 藏头诗模型
text_model3 = hub.Module(name="ernie_gen_couplet")   # 获取对联下句
def chat_bot(content, mode):
    res = ''
    if mode == '0':
        res = get_content(content)
        # print(res)
    elif mode == '1':# 清话
        res = text_model1.generate(texts=[content], use_gpu=False, beam_width=1)
        if res is None:
            return
        res = res[0][0]
        # print(res)
    elif mode == '2':#藏头诗
        res = text_model2.generate(texts=[content], use_gpu=False, beam_width=1)
        if res is None:
            return
        res = res[0][0]
    elif mode == '3':#对联
        res = text_model3.generate(texts=[content], use_gpu=False, beam_width=1)
        out = []
        res = res[0][0]
    return res

# 获取城市天气
def get_weather_data(city_name):
    weatherJsonUrl = "http://wthrcdn.etouch.cn/weather_mini?city={}".format(city_name)  # 将链接定义为一个字符串
    response = requests.get(weatherJsonUrl)  # 获取并下载页面，其内容会保存在respons.text成员变量里面
    response.raise_for_status()  # 这句代码的意思如果请求失败的话就会抛出异常，请求正常就上面也不会做
    # 将json文件格式导入成python的格式
    weather_dict = json.loads(response.text)
    # print(weather_dict)
    if weather_dict['desc'] == 'invilad-citykey':
        weather_info = '请输入正确的城市名!'
    else:
        forecast = weather_dict.get('data').get('forecast')
        city = '城市：' + weather_dict.get('data').get('city') + '\n'
        date = '日期：' + forecast[0].get('date') + '\n'
        type = '天气：' + forecast[0].get('type') + '\n'
        wendu = '温度：' + weather_dict.get('data').get('wendu') + '℃ ' + '\n'
        high = '高温：' + forecast[0].get('high') + '\n'
        low = '低温：' + forecast[0].get('low') + '\n'
        ganmao = '感冒提示：' + weather_dict.get('data').get('ganmao') + '\n'
        fengxiang = '风向：' + forecast[0].get('fengxiang')
        weather_info = city + date + type + wendu + high + low + ganmao + fengxiang
    return weather_info