老照片修复AI模型介绍及使用训练方法（五）

五、使用Flask3.0.0框架实现web的resful接口

1、安装flask框架和必要库

#安装flask框架
pip install flask
#安装mysql驱动pymysql
pip install pymysql
#安装ORM支持flask-sqlalchemy
pip install flask-sqlalchemy
#安装flask允许跨域工具
pip install flask_cors

2、创建项目并创建必要目录

├── app.py                     # 入口启动文件
├── settings.py                # 配置文件（自己创建）
├── model                      # AI模型包（自己创建）
├── static                     # 静态资源目录
├── templates                  # 模板文件目录

3、app.py下写上传接口并调用模型

@app.route('/upload', methods=['POST'])
def upload_file():
    # 获取图片文件
    img_file = request.files['file']
    model_name = request.form['model_name']
    if not img_file:
        # 如果没有文件则返回错误
        return jsonify({
            'msg': '未找到图片',
            'code': 0
        })
    # 将图片名按照 .  进行切分， 找到最后一个元素，也就是  文件的后缀名
    end_prefix = img_file.filename.split('.', 1)[1]
    # 通过文件的后缀名判断 身份为 合法的  图片
    if end_prefix.lower() not in current_app.config.get('IMG_ALLOW'):
        return jsonify({
            'msg': '图片格式不允许',
            'code': 0
        })
    server_img_upload = current_app.config.get('SERVER_IMG_UPLOAD')
    filename = str(uuid4()) + '.' + end_prefix  # 为了生成一个不重复的文件名
    outputFilename = filename.split('.')[0] + '_1.png'  # 新突破名称
    img_path = os.path.join(server_img_upload, filename)
    out_img_path = os.path.join(server_img_upload, outputFilename)
    img_file.save(img_path)
    # 以下是抠图模型使用

4.本地运行

python app.py --host=0.0.0.0 --port=8000
#启动后演示地址：http://localhost:8000/

未完待续...

作者：admin

TTF的家园-www.ttfde.top 个人博客以便写写东西，欢迎喜欢互联网的朋友一起交流!

TTF的家园 站点 QQ交谈