对于一些特殊的行业(比如交警需要统计车辆信息,小区物业可能需要统计出入信息等),需要监控视频人数统计的功能,而这样的功能一般由软件层面进行实现,接下来小编就带来一个使用flask框架,调用yolov5库进行人数统计的方法,并提供出一个http API供其他应用调用。
一、图示
客户端请求输入一段视频或者一个视频流,输出人数或其他目标数量,上报给上层服务器端,即提供一个http API调用算法统计出人数,最终http上报总人数
二、准备
相关技术 python pytorch opencv http协议 post请求
Flask
Flask是一个Python实现web开发的微框架,对于像我对web框架不熟悉的人来说还是比较容易上手的。
Flask安装
sudo pip install Flask
三、一个简单服务器应用
为了稍微了解一下flask是如何使用的,先做一个简单的服务器例子。
第一个文件hello.py。
from flask import Flask
app = Flask(__name__)
@app.route("/")
def hello():
return 'hello world!'
@app.route("/python")
def hello_python():
return 'hello python!'
if __name__ == '__main__':
app.run(host='0.0.0.0')
app.run(host=‘0.0.0.0')表示现在设定的ip为0.0.0.0,并且设定为0.0.0.0是非常方便的,如果你是在一台远程电脑上设置服务器,并且那台远程电脑的ip是172.1.1.1,那么在本地的电脑上可以设定ip为172.1.1.1来向服务器发起请求。
@app.route('/')表示发送request的地址是http://0.0.0.0:5000/,@app.route(