创新实训第三周团队博客

创新实训第三周团队博客

这一周主要完成了剩余的App前端代码和后端接口，PSGAN模型的微调和flask上妆接口。具体工作参考每个团队成员的个人博客。

前端App工作

整合后端接口和PSGAN的上妆接口，修复App的一些bug。

最终效果如下所示：

后端工作

和前端协调接口开发，实现动态相关接口，部分文档如下：

１，发布动态（需要验权）
１）方法：POST
２）URL：/star/publish
３）参数：content动态内容, images动态图片，目前只支持一张
４）响应：内部错误返回状态码 500，发布成功返回状态码 200,并携带动态信息

２，转发动态（需要验权）
１）方法：POST
２）URL：/star/forward/:starID
３）参数：content和images同上，starID是转发动态的id
４）响应：内部错误返回状态码 500，发布成功返回状态码 200,并携带动态信息

３，更新动态（需要验权）
１）方法：POST
２）URL：/star/update/:starID
３）参数：参数同１
４）响应：内部错误返回状态码 500，发布成功返回状态码 200,并携带动态信息

４，删除动态（需要验权）
１）方法：GET
２）URL：/star/delete/:starID
３）参数：starID是动态id
４）响应：内部错误返回状态码 500，发布成功返回状态码 200

５，点赞动态（需要验权）
１）方法：GET
２）URL：/star/like/:starID
３）参数：starID是动态id
４）响应：内部错误返回状态码 500，发布成功返回状态码 200

６，取消点赞动态（需要验权）
１）方法：GET
２）URL：/star/cancel_like/:starID
３）参数：starID是动态id
４）响应：内部错误返回状态码 500，发布成功返回状态码 200

７，动态详情
１）方法：GET
２）URL：/star/details/:starID
３）参数：starID是动态id
４）响应：内部错误返回状态码 500，动态已被删除返回状态码400，发布成功返回状态码 200，携带动态详情数据

８，获取点赞用户列表
１）方法：GET
２）URL：/star/like_users/:starID
３）参数：starID是动态id
４）响应：内部错误返回状态码 500，发布成功返回状态码 200，携带点赞用户列表

９，获取转发动态列表
１）方法：GET
２）URL：/star/forward_stars/:starID
３）参数：starID是动态id
４）响应：内部错误返回状态码 500，发布成功返回状态码 200，携带转发动态列表

10，获取用户全部动态
１）方法：GET
２）URL：/user/:id/stars
３）参数：id是用户id
４）响应：内部错误返回状态码 500，发布成功返回状态码 200，携带动态列表

PSGAN 模型搭建

通过flask框架完成模型上妆接口。

flask相关代码如下：

from flask import Flask, request, render_template, Response

from data_loaders.makeup_utils import *
from solver_psgan import Solver_PSGAN

app = Flask(__name__)


@app.route('/generate_makeup', methods=['POST', 'GET'])
def generate_makeup():
    if request.method == 'POST':
        try:
            example_image = request.files.get('example_image')
            user_image = request.files.get('user_image')
            shade_alpha = request.form.get("shade_alpha")
            local_flag = request.form.get("local_flag")
            lip_flag = request.form.get("lip_flag")
            eye_flag = request.form.get("eye_flag")
            face_flag = request.form.get("face_flag")

            if str(local_flag) == '1':
                print('local flag')
                images = [user_image, example_image]
                mask2use = get_mask(Image.open(images[1]), lip_flag, eye_flag, face_flag)
                images = [Image.open(image) for image in images]
                images = [preprocess_image(image) for image in images]
                transferred_image = Solver_PSGAN.partial_test(*(images[0]), *(images[1]), *(images[0]), mask2use, shade_alpha=float(shade_alpha))
            else:
                images = [user_image, example_image]
                images = [Image.open(image) for image in images]
                images = [preprocess_image(image) for image in images]
                transferred_image = Solver_PSGAN.image_test(*(images[0]), *(images[1]), shade_alpha=float(shade_alpha))
            transferred_image.save('./static/images/transferred_image.png')

            with open('static/images/transferred_image.png', 'rb') as f:
                image = f.read()
                resp = Response(image, status=200, mimetype="image/png")
            return resp
        except Exception as e:
            print(str(e))
            resp = Response(response='Exception', status=500, content_type='text/html;charset=utf8')
            return resp

    return render_template('index.html')


if __name__ == '__main__':
    app.run(debug=True)

用于测试接口的HTML页面如下：