<返回更多

使用FastAPI部署YOLO模型的步骤

2023-12-06  微信公众号  小白玩转Python
加入收藏

在计算机视觉领域,You Only Look Once (YOLO) 算法已经崭露头角,成为一种改变游戏规则的算法。它承诺具有卓越准确性的实时目标检测,使其成为从监视和自动驾驶车辆到图像和视频分析等应用中强大的工具。然而,只有在无缝集成到实际的现实系统中时,YOLO 的真正潜力才能被充分发挥。这就是现代、快速、用于使用Python/ target=_blank class=infotextkey>Python构建API的Web框架FastAPI的用武之地,它可以轻松地成为您在部署YOLO模型时的伙伴。

使用FastAPI部署YOLO模型的步骤

想象一下能够在Web应用程序中部署一个YOLO模型,允许用户通过简单的API调用进行实时目标检测。无论您是构建智能安全系统、野生动物监测应用程序还是零售分析平台,本指南将引导您完成整个过程,从设置开发环境到使用FastAPI部署完全功能的YOLO模型。

在深入研究本教程时,您将揭示YOLO的魔力——它如何能够在眨眼之间识别图像和视频中的物体。您还将掌握将这种强大算法与FastAPI集成的艺术,这是为那些注重速度和简单性的开发人员设计的框架。通过本次旅程的结束,您将具备创建自己的实时目标检测API的工具和知识,这些API可以部署在云端、本地服务器甚至边缘设备上。

因此,无论您是经验丰富的计算机视觉工程师还是渴望探索YOLO和FastAPI世界的好奇开发人员,系好安全带吧。我们将开始一场逐步部署YOLO模型的专业之旅。准备将您的目标检测梦想变为现实吗?让我们开始吧!

使用FastAPI部署YOLO模型的步骤

第一部分:设置环境

在我们深入使用FastAPI部署YOLO模型的世界之前,我们需要确保我们的开发环境已正确设置。本节将逐步引导您完成这个过程。

1. 安装Python

首先,请确保您的系统上已安装Python。您可以从官方网站:https://www.python.org/downloads/下载最新版本的Python,或使用Anaconda等包管理器。要检查Python是否已安装,请打开终端或命令提示符运行:

python - version

2. 创建虚拟环境

为了保持项目的依赖关系隔离,最好创建一个虚拟环境。这样,您就可以避免不同项目之间的冲突。让我们使用Python的内置 venv 模块创建一个虚拟环境。打开终端并导航到项目的根目录。运行以下命令:

# Create a virtual environment (replace 'myenv' with your preferred environment name)
python -m venv myenv
# Activate the virtual environment (windows)
myenvScriptsactivate
# Activate the virtual environment (macOS/linux)
source myenv/bin/activate

您应该看到终端提示更改,指示虚拟环境处于活动状态。

注意:请记得将'myenv'替换为您喜欢的虚拟环境名称。

3. 安装依赖

现在,您正在虚拟环境中工作,是时候安装必要的依赖项了。这些包括FastAPI、Uvicorn(用于提供FastAPI应用程序的工具)、与YOLO相关的库以及您可能需要的项目的任何其他包。使用 pip 安装这些依赖项:

pip install fastapi uvicorn opencv-python-headless numpy

4. YOLO模型设置

要使用YOLO,您需要安装来自ultralytics的yolov8库。

pip install ultralytics

有了开发环境的设置,您现在已经准备好深入研究YOLO和FastAPI的激动人心的世界。在接下来的部分中,我们将探讨如何利用YOLO的强大功能进行实时目标检测,并构建一个用于提供服务的FastAPI应用程序。

第二部分:创建FastAPI应用程序

现在是时候动手构建一个FastAPI应用程序来部署模型了。本节将引导您完成设置对象检测API基础的过程。

1. 项目结构

让我们开始整理我们的项目结构。创建一个用于FastAPI项目的目录并进入其中:

mkdir object_detection_api
cd object_detection_api

在这个项目目录中,您将为FastAPI应用程序的不同组件创建文件和文件夹。

2. 初始化FastAPI应用程序

FastAPI让构建Web应用程序变得非常容易。创建一个用于FastAPI应用程序的Python脚本,通常命名为 mAIn.py :

touch main.py

现在,让我们开始编写一些代码。在您喜欢的文本编辑器或IDE中打开 main.py,并导入FastAPI,现在FastAPI应用程序初始化完成。这个应用程序将作为您的目标检测API的基础。

3. 创建您的第一个路由

在FastAPI中,您使用Python函数定义路由。让我们从一个简单的“Hello, World!”路由开始。将以下代码添加到 main.py

@App.get("/")
async def read_root():
return {"message": "Hello, World!"}

这段代码定义了一个路由,响应根URL(“/”)的GET请求,并返回一个带有“message”字段的JSON响应。

4. 本地运行FastAPI应用程序

现在,是时候在本地测试您的FastAPI应用程序了。打开终端并导航到包含 main.py 的项目目录。如果还没有激活虚拟环境,请激活:

source myenv/bin/activate # Replace 'myenv' with your environment name

接下来,使用Uvicorn运行您的FastAPI应用程序:

uvicorn main:app --reload

这个命令告诉Uvicorn从 main.py 模块运行 app 对象,并启用开发环境下的自动重新加载。您应该看到输出,指示您的FastAPI应用程序正在本地运行。默认情况下,它在 http://127.0.0.1:8000 上运行。

5. 访问Hello World路由

打开您的Web浏览器或使用 curl 等工具访问“Hello, World!”路由:

curl http://127.0.0.1:8000/

您应该收到一个带有“Hello, World!”消息的JSON响应。有了您的FastAPI应用程序运行起来,您现在可以继续进行激动人心的部分:集成YOLOv8模型进行目标检测。在接下来的部分中,我们将探讨如何准备YOLOv8模型,并将其与FastAPI无缝集成。

第三部分:将YOLOv8与FastAPI集成

现在我们已经有了FastAPI应用程序,让我们深入研究如何集成YOLOv8模型进行实时目标检测的过程。本节将引导您完成无缝将YOLOv8与FastAPI结合的步骤。

1. 加载YOLOv8模型

让我们从在FastAPI应用程序中加载YOLOv8模型开始。打开 main.py 并在文件顶部添加以下代码以导入必要的模块:

import cv2
import numpy as np
from ultralytics import YOLO

2. 创建一个目标检测路由

现在,让我们在FastAPI中创建一个路由,该路由将接受一个用于目标检测的图像。定义一个新的路由函数如下:

from fastapi import File, UploadFile
model = YOLO("yolov8n.pt")

@app.post("/detect/")
async def detect_objects(file: UploadFile):
 # Process the uploaded image for object detection
 image_bytes = await file.read()
 image = np.frombuffer(image_bytes, dtype=np.uint8)
 image = cv2.imdecode(image, cv2.IMREAD_COLOR)
 
 # Perform object detection with YOLOv8
 detections = model.predict(image)
 
 return {"detections": detections}

在这里,我们创建了一个名为 /detect/ 的路由,该路由接受上传的图像文件。我们将使用 model.predict() 在上传的图像上执行目标检测。

3. 测试目标检测路由

完成了YOLOv8的集成,现在您可以测试您的目标检测路由。使用Uvicorn启动您的FastAPI应用程序:

uvicorn main:app --reload

然后,向/detect/路由发出POST请求,上传一个图像文件进行目标检测。您可以使用curl或Postman等工具进行此操作。

curl -X POST -F "file=@image.jpg" http://127.0.0.1:8000/detect/

您将收到一个带有目标检测结果的JSON响应。恭喜!您已成功将YOLOv8与FastAPI集成,实现了实时目标检测。在接下来的部分中,我们将增强API,添加文档,并探索部署选项。

第四部分:部署FastAPI应用程序

现在您已经构建了FastAPI应用程序,是时候将其部署,使您的目标检测API可以被用户访问了。在本节中,我们将探讨各种部署选项,包括用于测试的本地部署和用于生产环境的基于云的部署。

1. 本地部署进行测试

在部署到生产环境之前,测试FastAPI应用程序在本地是否正常运行是必不可少的。要在本地运行FastAPI应用程序,请打开终端,导航到包含 main.py 的项目目录,并激活虚拟环境(如果尚未激活):

source myenv/bin/activate # Replace 'myenv' with your environment name

然后,使用Uvicorn启动FastAPI应用程序:

uvicorn main:app --reload

您的FastAPI应用程序现在应该在 http://127.0.0.1:8000/ 上可访问。您可以使用 curl 、Postman或您的Web浏览器测试API端点。

2. 用于生产的基于云的部署

当您准备将FastAPI应用程序部署到生产环境时,您有几个基于云的部署选项。一些流行的选择包括:

具体的部署方法可能取决于您选择的云提供商。通常需要:

3. 选择正确的服务器

在部署到基于云的服务器时,您可能可以灵活选择服务器类型。常见的选项包括:

4. 持续集成和持续部署(CI/CD)

考虑实施CI/CD流水线以自动化部署过程。Jenkins、Travis CI、GitLab CI/CD和Github Actions等工具可以帮助您在将更改推送到代码仓库时自动进行测试和部署。通过遵循CI/CD的最佳实践,您可以确保平稳可靠的部署过程,降低在生产环境中出现错误的风险。

关键词:FastAPI      点击(9)
声明:本站部分内容来自互联网,如有版权侵犯或其他问题请与我们联系,我们将立即删除或处理。
▍相关推荐
更多FastAPI相关>>>