yolov5部署web服务

计算机CV民工

已于 2023-02-22 01:11:46 修改

阅读量1.1k

点赞数

分类专栏：缺陷检测文章标签： python

于 2023-02-20 13:41:38 首次发布

本文链接：https://blog.csdn.net/qq_41701723/article/details/129121763

版权

缺陷检测专栏收录该内容

40 篇文章 0 订阅

订阅专栏

yolov5部署web服务

上传图片
- 注意
上传视频
效果
pycharm搜索
Pycharm上一个断点调到下一个断点

上传图片

一段上传和访问的demo

#!/usr/bin/python3
#encoding:utf-8
#来自官网,但官网的不能直接用,稍微修改了一下,秒秒钟运行成功,哈哈哈哈
import os
from flask import Flask, request, redirect, url_for
from werkzeug.utils import secure_filename
from flask import send_from_directory
UPLOAD_FOLDER = './inference/images'#上传到这里
ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif','jpeg','rar','doc','docx','dot','pptx','db'])#允许的格式,保证安全性
app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
app.config['MAX_CONTENT_LENGTH'] = 64 * 1024 * 1024#限制大小64mb
def allowed_file(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS


@app.route('/', methods=['GET', 'POST'])
def upload_file():
    if request.method == 'POST':
        file = request.files['file']
        if file and allowed_file(file.filename):
            #filename = secure_filename(file.filename)
            filename =request.form['filename']#文件名,支持中文哦！！！
            filename =str(filename)#防止恶意传送非正常字符导致服务器异常
            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
            return redirect(url_for('uploaded_file',
                                    filename=filename))
    return '''
    <!doctype html>
    <title>上传文件</title>
    <h1>上传文件</h1>
    <form action="" method=post enctype=multipart/form-data>
      <p><input type=file name=file></br>
      <input type="text" name="filename"></p>
         <input type=submit value="上传">
    </form>
    '''

@app.route('/uploads/<filename>')
def uploaded_file(filename):
    return send_from_directory(app.config['UPLOAD_FOLDER'],
                               filename)

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