yolov5-fastapi-demo
这是一个演示的FastAPI应用程序,允许用户上传图像,使用预先训练的YOLOv5模型执行推理以及接收JSON格式的结果。 此存储库还包括Jinja2 HTML模板,因此您可以通过Web浏览器(位于localhost:8000)访问此界面。
要求
安装了所有require.txt依赖关系的Python 3.8或更高版本,包括torch> = 1.7(根据 )。
要安装运行:
pip install -r requirements.txt
推理
您可以使用python server.py或uvicorn server:app --reload初始化服务器
您可以通过以下几种方法测试服务器:
使用client.py这是使用请求库将一批图像和模型名称上载到localhost:8000/detect/并接收JSON推理结果的基本示例。
在网络浏览器
2021-06-06 15:06:07
651KB
Python
1