五、使用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/
未完待续...
- 本文固定链接: http://www.ttfde.top/index.php/post/433.html
- 转载请注明: admin 于 TTF的家园 发表
《本文》有 0 条评论