DAMOYOLO-S集成JavaScript前端:打造交互式Web目标检测Demo
2026/6/12 3:15:51 网站建设 项目流程

DAMOYOLO-S集成JavaScript前端:打造交互式Web目标检测Demo

1. 引言

你有没有想过,把一个强大的目标检测模型,变成一个在浏览器里就能直接玩的工具?比如上传一张街景照片,网页上立刻就能框出所有的车辆和行人;或者打开摄像头,实时视频里就能看到动态的检测框跟着物体移动。这听起来像是专业软件才能做的事,但其实,用我们手头的DAMOYOLO-S模型和一些基础的Web技术,就能轻松实现。

对于很多开发者来说,模型训练和部署是一回事,但怎么把它变成一个用户能直观感受和使用的产品,又是另一道坎。后端模型推理的代码写好了,但用户总不能对着命令行窗口操作。这时候,一个轻量、交互友好的Web前端就显得至关重要。它不仅是模型的“脸面”,更是连接复杂AI能力与普通用户之间的桥梁。

本文将带你一步步走通这个流程:从搭建一个提供检测API的Python后端开始,到用纯JavaScript构建一个能够上传图片、处理视频流并实时渲染检测结果的前端页面。整个过程,我们聚焦于“可用”和“易懂”,用最直接的代码展示如何将DAMOYOLO-S的检测能力,无缝集成到一个现代化的Web应用中。无论你是想快速验证模型效果,还是希望构建一个AI应用的原型,这个方案都能提供一个坚实的起点。

2. 项目整体架构与思路

在动手写代码之前,我们先理清整个Demo要做什么,以及各个部分如何协同工作。这样在开发时,思路会更清晰。

我们的目标是构建一个B/S(浏览器/服务器)架构的Web应用。核心工作流很简单:用户在浏览器前端选择图片或开启摄像头,前端将图像数据发送到我们的后端服务器,后端调用DAMOYOLO-S模型进行推理,然后将检测结果(包括物体类别、位置坐标)返回给前端,最后由前端在原始图像上绘制出检测框和标签,展示给用户。

2.1 技术栈选型

为了实现上述流程,我们需要选择合适的技术:

  • 后端 (Server-side):

    • 框架: 选用Flask。它足够轻量、灵活,对于构建RESTful API快速原型来说非常合适。当然,如果你追求更高的性能,FastAPI是更现代的替代品,其异步特性和自动API文档生成也很吸引人。本文以Flask为例进行讲解,其思想同样适用于FastAPI。
    • 核心模型:DAMOYOLO-S。这是一个在速度和精度上平衡得较好的目标检测模型,适合需要实时反馈的Web应用场景。
    • 推理库:PyTorchONNX Runtime。取决于你最终部署的模型格式。
  • 前端 (Client-side):

    • 核心语言:Vanilla JavaScript (原生JS)。我们不依赖React、Vue等重型框架,以最纯粹的方式展示如何通过JavaScript的Fetch APICanvasVideo元素与后端交互,这使得代码更易于理解和移植。
    • UI 与 样式: 简单的HTML5CSS3。构建一个包含文件上传、实时视频和结果显示区域的基础界面。
  • 通信桥梁:

    • API 接口: 遵循RESTful风格设计。前端通过HTTPPOST请求将图像数据发送到后端的某个接口(例如/detect)。
    • 数据格式: 使用JSON作为前后端数据交换的标准格式。前端发送Base64编码的图片或帧数据,后端返回结构化的检测结果。

2.2 工作流程拆解

整个应用跑起来,就像一场精心安排的接力赛:

  1. 用户交互: 用户在网页上点击“选择图片”按钮,或者点击“开启摄像头”。
  2. 数据采集与预处理:
    • 如果是图片,前端读取文件并将其转换为Base64字符串或FormData
    • 如果是摄像头,前端使用getUserMedia获取视频流,并定期(例如每秒10帧)将当前视频帧捕获到Canvas上,再转换为图像数据。
  3. 网络请求: 前端使用JavaScript的fetch函数,将图像数据打包,发送到后端的检测API(如http://your-server-address:5000/detect)。
  4. 模型推理: 后端Flask应用接收到请求。
    • 从请求中解析出图像数据,并解码为OpenCV或PIL可以处理的格式。
    • 调用预先加载好的DAMOYOLO-S模型对图像进行推理。
    • 对模型的原始输出进行后处理,包括非极大值抑制(NMS),得到最终的边界框、置信度和类别ID。
  5. 结果返回: 后端将处理后的结果(一个包含多个检测框信息的列表)封装成JSON格式,返回给前端。
  6. 结果渲染: 前端收到JSON响应后,解析数据。
    • 在另一个Canvas画布上,先绘制原始图像。
    • 然后遍历每一个检测结果,根据其坐标信息,在画布上绘制矩形框。
    • 在框的附近,用文字标注出检测到的物体类别和置信度。
  7. 实时循环(针对视频): 如果是视频流模式,则第2至第6步会在一个requestAnimationFramesetInterval循环中不断执行,从而实现实时检测的效果。

理清了这些,我们就可以开始分步实现了。接下来,我们先从后端服务搭建做起。

3. 后端服务搭建:Flask与DAMOYOLO-S模型API

后端是整个应用的大脑,它负责加载模型、处理请求、运行推理。我们首先来构建这个“大脑”。

3.1 环境准备与依赖安装

建议使用Python虚拟环境来管理项目依赖,避免包冲突。创建一个新的项目目录,并在其中初始化环境。

# 创建项目目录 mkdir damoyolo-web-demo && cd damoyolo-web-demo # 创建虚拟环境 (以venv为例) python -m venv venv # 激活虚拟环境 # Windows: venv\Scripts\activate # Linux/Mac: source venv/bin/activate

激活虚拟环境后,安装必要的Python包。我们主要需要Flask来处理Web请求,以及PyTorch和OpenCV来处理图像和模型推理。

# 安装核心依赖 pip install flask torch torchvision opencv-python pillow numpy # 如果需要使用DAMOYOLO官方仓库,可能还需要安装其依赖 # pip install damo-yolo # 请根据官方仓库说明安装

3.2 构建Flask应用与检测API

首先,我们创建一个名为app.py的文件,这是后端应用的入口。

# app.py import io import base64 import json from flask import Flask, request, jsonify import cv2 import numpy as np from PIL import Image import torch from torchvision import transforms # 假设我们已经有一个加载好的DAMOYOLO-S模型和预处理、后处理函数 # 这里我们先定义它们的接口,具体实现取决于你如何获取和使用模型 def load_model(): """加载DAMOYOLO-S模型""" # 示例:从本地文件加载 # model = torch.hub.load('path/to/damoyolo', 'damoyolo_s', pretrained=True, source='local') # model.eval() # return model print("模型加载函数被调用。在实际应用中,请在此处实现模型加载逻辑。") return None def preprocess_image(image_np): """将numpy图像数组预处理为模型输入张量""" # 示例:调整大小、归一化、转换为Tensor # transform = transforms.Compose([...]) # input_tensor = transform(Image.fromarray(image_np)).unsqueeze(0) # return input_tensor print("图像预处理函数被调用。") return torch.randn(1, 3, 640, 640) # 返回一个模拟的张量 def postprocess_output(model_output, orig_img_shape): """将模型输出后处理为边界框、置信度、类别列表""" # 示例:应用NMS,过滤低置信度框,将坐标映射回原图尺寸 # boxes = [...] # scores = [...] # class_ids = [...] # return boxes, scores, class_ids print("输出后处理函数被调用。") # 返回一些模拟数据用于测试前端 boxes = [[100, 100, 200, 200], [300, 150, 400, 300]] scores = [0.95, 0.87] class_ids = [0, 2] # 假设0代表人,2代表车 class_names = {0: 'person', 2: 'car'} # 类别映射 return boxes, scores, class_ids, class_names app = Flask(__name__) # 全局加载模型 (在实际应用中,注意线程安全) model = load_model() @app.route('/') def index(): return "DAMOYOLO-S Detection API is running." @app.route('/detect', methods=['POST']) def detect(): """ 检测API接口。 接收JSON格式请求,其body中应包含一个'image'字段,值为Base64编码的图片字符串(不含data:image前缀)。 返回JSON格式的检测结果。 """ if not request.is_json: return jsonify({'error': 'Request must be JSON'}), 400 data = request.get_json() image_b64 = data.get('image') if not image_b64: return jsonify({'error': 'No image data provided'}), 400 try: # 1. 解码Base64图像 image_data = base64.b64decode(image_b64) image_np = np.frombuffer(image_data, np.uint8) image_np = cv2.imdecode(image_np, cv2.IMREAD_COLOR) if image_np is None: return jsonify({'error': 'Invalid image data'}), 400 # 可选:将BGR转换为RGB(取决于模型训练时的通道顺序) # image_rgb = cv2.cvtColor(image_np, cv2.COLOR_BGR2RGB) orig_h, orig_w = image_np.shape[:2] # 2. 预处理 input_tensor = preprocess_image(image_np) # 3. 模型推理 (这里用模拟推理代替) # with torch.no_grad(): # outputs = model(input_tensor) print("执行模型推理(模拟)...") # 4. 后处理 boxes, scores, class_ids, class_names = postprocess_output(None, (orig_h, orig_w)) # 5. 格式化结果 detections = [] for box, score, cls_id in zip(boxes, scores, class_ids): x1, y1, x2, y2 = box detections.append({ 'bbox': [float(x1), float(y1), float(x2), float(y2)], # 转为float便于JSON序列化 'score': float(score), 'class_id': int(cls_id), 'class_name': class_names.get(cls_id, f'class_{cls_id}') }) # 6. 返回结果 return jsonify({ 'success': True, 'detections': detections, 'image_shape': {'height': orig_h, 'width': orig_w} }) except Exception as e: print(f"Detection error: {e}") return jsonify({'error': str(e)}), 500 if __name__ == '__main__': # 在本地开发时运行,host='0.0.0.0'允许局域网访问,debug=True便于调试 app.run(host='0.0.0.0', port=5000, debug=True)

代码要点解释

  1. 模型加载与处理函数(load_model,preprocess_image,postprocess_output): 这些是核心的AI部分。你需要根据DAMOYOLO-S模型的具体使用方式来实现它们。示例中使用了打印语句和模拟数据,确保API流程能跑通,方便我们先测试前端。
  2. Flask应用与路由:
    • @app.route('/'): 根路径,用于简单验证服务是否启动。
    • @app.route('/detect', methods=['POST']): 核心的检测接口。它只接受POST请求和JSON格式的数据。
  3. 请求处理流程:
    • 验证: 检查请求是否为JSON,并包含image字段。
    • 解码: 将Base64字符串解码为原始的字节数据,然后用OpenCV解码成NumPy数组。
    • 推理: 调用预处理、模型推理、后处理函数。注意:示例中推理是模拟的,你需要替换为真实的模型调用。
    • 格式化: 将检测结果组织成一个字典列表,每个字典代表一个检测到的物体。
    • 返回: 将结果以JSON格式返回,包含成功标志、检测列表和原图尺寸。

3.3 运行与测试后端API

保存好app.py后,在终端运行它:

python app.py

如果看到类似* Running on http://0.0.0.0:5000的输出,说明后端服务已经启动。

现在,我们可以用任何HTTP客户端(如curl、Postman)来测试这个API。这里提供一个简单的Python测试脚本:

# test_api.py import requests import base64 import json # 1. 读取一张测试图片并编码为Base64 with open('test.jpg', 'rb') as f: # 请准备一张名为test.jpg的图片 img_b64 = base64.b64encode(f.read()).decode('utf-8') # 2. 构造请求数据 url = 'http://127.0.0.1:5000/detect' headers = {'Content-Type': 'application/json'} data = {'image': img_b64} # 3. 发送POST请求 response = requests.post(url, headers=headers, data=json.dumps(data)) # 4. 打印响应 print(f"Status Code: {response.status_code}") print(f"Response JSON: {response.json()}")

运行这个测试脚本,你应该会收到一个包含模拟检测结果的JSON响应。这表明后端API已经就绪,可以接受前端的调用了。

4. 前端开发:使用JavaScript构建交互界面

后端准备好了,现在我们来打造用户直接看到和操作的“脸面”。我们将创建一个简单的HTML页面,并用原生JavaScript实现所有交互逻辑。

4.1 基础HTML结构与样式

创建一个index.html文件,构建页面的骨架和基础样式。

<!DOCTYPE html> <html lang="zh-CN"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <title>DAMOYOLO-S 交互式目标检测 Demo</title> <style> * { box-sizing: border-box; margin: 0; padding: 0; font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; } body { background-color: #f5f7fa; color: #333; line-height: 1.6; padding: 20px; max-width: 1200px; margin: 0 auto; } header { text-align: center; margin-bottom: 30px; padding-bottom: 20px; border-bottom: 2px solid #e1e4e8; } h1 { color: #2c3e50; margin-bottom: 10px; } .subtitle { color: #7f8c8d; font-size: 1.1em; } .container { display: flex; flex-wrap: wrap; gap: 30px; margin-bottom: 30px; } .panel { background: white; border-radius: 10px; padding: 25px; box-shadow: 0 5px 15px rgba(0, 0, 0, 0.08); flex: 1; min-width: 300px; } .panel h2 { color: #3498db; margin-bottom: 20px; padding-bottom: 10px; border-bottom: 1px solid #eee; } .controls { display: flex; flex-direction: column; gap: 20px; } .button-group { display: flex; flex-wrap: wrap; gap: 10px; margin-top: 10px; } button, input[type="file"] { padding: 12px 24px; border: none; border-radius: 6px; cursor: pointer; font-size: 16px; font-weight: 600; transition: all 0.3s ease; } button { background-color: #3498db; color: white; flex: 1; min-width: 140px; } button:hover { background-color: #2980b9; transform: translateY(-2px); } button:active { transform: translateY(0); } button#stopCam { background-color: #e74c3c; } button#stopCam:hover { background-color: #c0392b; } input[type="file"] { background-color: #ecf0f1; padding: 10px; width: 100%; } .canvas-container { text-align: center; margin-top: 15px; } canvas { max-width: 100%; border-radius: 8px; box-shadow: 0 3px 10px rgba(0,0,0,0.1); border: 1px solid #ddd; } #resultCanvas { border-color: #3498db; } .status { margin-top: 20px; padding: 15px; border-radius: 6px; background-color: #f8f9fa; border-left: 4px solid #3498db; font-family: monospace; white-space: pre-wrap; word-break: break-all; min-height: 60px; max-height: 200px; overflow-y: auto; } .status.error { border-left-color: #e74c3c; background-color: #fdf2f2; } .status.success { border-left-color: #2ecc71; background-color: #f0f9f4; } footer { text-align: center; margin-top: 40px; color: #95a5a6; font-size: 0.9em; padding-top: 20px; border-top: 1px solid #e1e4e8; } @media (max-width: 768px) { .container { flex-direction: column; } } </style> </head> <body> <header> <h1>🚀 DAMOYOLO-S 交互式目标检测 Demo</h1> <p class="subtitle">上传图片或开启摄像头,体验实时目标检测</p> </header> <div class="container"> <div class="panel"> <h2>📁 图片检测</h2> <div class="controls"> <input type="file" id="fileInput" accept="image/*"> <div class="button-group"> <button id="detectImage">检测图片</button> <button id="clearImage">清空画布</button> </div> <div class="canvas-container"> <p><strong>检测结果:</strong></p> <canvas id="resultCanvas"></canvas> </div> </div> </div> <div class="panel"> <h2>📹 实时视频检测</h2> <div class="controls"> <div class="button-group"> <button id="startCam">开启摄像头</button> <button id="stopCam" disabled>停止检测</button> </div> <div class="canvas-container"> <p><strong>实时画面:</strong></p> <video id="videoElement" autoplay playsinline style="display:none; width:100%; border-radius:8px;"></video> <canvas id="videoCanvas"></canvas> </div> <div> <label for="fpsSlider">检测频率 (FPS): </label> <input type="range" id="fpsSlider" min="1" max="30" value="10"> <span id="fpsValue">10</span> </div> </div> </div> </div> <div class="panel"> <h2>📊 检测结果与状态</h2> <div id="statusBox" class="status"> 等待操作...请选择图片或开启摄像头。 </div> </div> <footer> <p>本Demo展示了DAMOYOLO-S模型与JavaScript前端的集成。后端API运行于 <code id="apiEndpoint">http://localhost:5000</code></p> <p>你可以修改下方配置以连接到不同的后端服务器。</p> <div style="margin-top: 10px;"> <label for="apiUrlInput">后端API地址: </label> <input type="text" id="apiUrlInput" value="http://localhost:5000/detect" style="width: 300px; padding: 5px;"> <button id="updateApiBtn" style="padding: 5px 15px; font-size: 14px;">更新</button> </div> </footer> <script src="main.js"></script> </body> </html>

这个HTML页面创建了一个双栏布局,左侧用于图片上传和检测,右侧用于实时视频检测。它包含了所有必要的UI元素:文件输入、按钮、视频元素、画布以及状态显示区域。

4.2 JavaScript核心逻辑实现

现在,创建main.js文件,这是前端所有交互逻辑的核心。我们将实现图片处理、视频流捕获、与后端API通信以及结果绘制。

// main.js document.addEventListener('DOMContentLoaded', function() { // --- 全局变量与DOM元素 --- const fileInput = document.getElementById('fileInput'); const detectImageBtn = document.getElementById('detectImage'); const clearImageBtn = document.getElementById('clearImage'); const startCamBtn = document.getElementById('startCam'); const stopCamBtn = document.getElementById('stopCam'); const videoElement = document.getElementById('videoElement'); const videoCanvas = document.getElementById('videoCanvas'); const resultCanvas = document.getElementById('resultCanvas'); const statusBox = document.getElementById('statusBox'); const fpsSlider = document.getElementById('fpsSlider'); const fpsValue = document.getElementById('fpsValue'); const apiUrlInput = document.getElementById('apiUrlInput'); const updateApiBtn = document.getElementById('updateApiBtn'); let videoStream = null; let detectionInterval = null; let currentFps = 10; let apiEndpoint = apiUrlInput.value; // 获取画布上下文 const resultCtx = resultCanvas.getContext('2d'); const videoCtx = videoCanvas.getContext('2d'); // --- 工具函数 --- function updateStatus(message, isError = false, isSuccess = false) { statusBox.textContent = message; statusBox.className = 'status'; if (isError) { statusBox.classList.add('error'); } else if (isSuccess) { statusBox.classList.add('success'); } console.log(message); } function drawDetection(ctx, detections, imgWidth, imgHeight, canvasWidth, canvasHeight) { // 计算缩放比例,使图像适应画布 const scaleX = canvasWidth / imgWidth; const scaleY = canvasHeight / imgHeight; const scale = Math.min(scaleX, scaleY); // 保持宽高比 // 计算在画布上居中的偏移量 const offsetX = (canvasWidth - imgWidth * scale) / 2; const offsetY = (canvasHeight - imgHeight * scale) / 2; // 绘制检测框和标签 detections.forEach(det => { const [x1, y1, x2, y2] = det.bbox; const score = det.score; const className = det.class_name; // 映射坐标到画布 const drawX1 = offsetX + x1 * scale; const drawY1 = offsetY + y1 * scale; const drawX2 = offsetX + x2 * scale; const drawY2 = offsetY + y2 * scale; const width = drawX2 - drawX1; const height = drawY2 - drawY1; // 绘制矩形框 ctx.strokeStyle = '#00FF00'; // 绿色框 ctx.lineWidth = 2; ctx.strokeRect(drawX1, drawY1, width, height); // 绘制标签背景 const label = `${className} ${(score * 100).toFixed(1)}%`; ctx.font = '16px Arial'; const textWidth = ctx.measureText(label).width; ctx.fillStyle = '#00FF00'; ctx.fillRect(drawX1, drawY1 - 20, textWidth + 10, 20); // 绘制标签文字 ctx.fillStyle = '#000'; ctx.fillText(label, drawX1 + 5, drawY1 - 5); }); } // --- 图片检测相关函数 --- function handleImageDetection(imageDataUrl) { updateStatus('正在发送图片进行检测...'); // 移除Base64 URL前缀 const base64Data = imageDataUrl.replace(/^data:image\/(png|jpeg|jpg);base64,/, ''); fetch(apiEndpoint, { method: 'POST', headers: { 'Content-Type': 'application/json', }, body: JSON.stringify({ image: base64Data }) }) .then(response => { if (!response.ok) { throw new Error(`HTTP error! status: ${response.status}`); } return response.json(); }) .then(data => { if (data.success) { updateStatus(`检测成功!共发现 ${data.detections.length} 个目标。`, false, true); // 在结果画布上绘制 const img = new Image(); img.onload = () => { // 设置画布尺寸与图片原始尺寸一致(或按需缩放) resultCanvas.width = img.width; resultCanvas.height = img.height; resultCtx.clearRect(0, 0, resultCanvas.width, resultCanvas.height); resultCtx.drawImage(img, 0, 0); drawDetection(resultCtx, data.detections, img.width, img.height, resultCanvas.width, resultCanvas.height); }; img.src = imageDataUrl; // 使用原始DataURL加载图片 } else { updateStatus(`检测失败: ${data.error || '未知错误'}`, true); } }) .catch(error => { updateStatus(`请求失败: ${error.message}`, true); console.error('Detection error:', error); }); } // --- 视频检测相关函数 --- function startVideoDetection() { if (!videoStream) { updateStatus('请先开启摄像头。', true); return; } if (detectionInterval) { clearInterval(detectionInterval); } updateStatus(`开始实时检测 (${currentFps} FPS)...`, false, true); // 设置定时器,按指定FPS发送帧进行检测 detectionInterval = setInterval(captureAndDetectFrame, 1000 / currentFps); } function stopVideoDetection() { if (detectionInterval) { clearInterval(detectionInterval); detectionInterval = null; updateStatus('已停止实时检测。'); } } function captureAndDetectFrame() { if (videoElement.readyState !== videoElement.HAVE_ENOUGH_DATA) { return; } // 将当前视频帧绘制到videoCanvas上 videoCanvas.width = videoElement.videoWidth; videoCanvas.height = videoElement.videoHeight; videoCtx.drawImage(videoElement, 0, 0, videoCanvas.width, videoCanvas.height); // 将canvas内容转换为Base64 const imageDataUrl = videoCanvas.toDataURL('image/jpeg', 0.8); // 使用JPEG压缩以减少数据量 const base64Data = imageDataUrl.replace(/^data:image\/(png|jpeg|jpg);base64,/, ''); // 发送到后端API (注意:这里为了性能,没有等待响应就继续下一帧,实际可根据需求调整) fetch(apiEndpoint, { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify({ image: base64Data }) }) .then(response => response.json()) .then(data => { if (data.success) { // 在videoCanvas上绘制检测结果 const tempCtx = videoCanvas.getContext('2d'); // 先清除上一帧的绘制,但保留视频图像 tempCtx.clearRect(0, 0, videoCanvas.width, videoCanvas.height); tempCtx.drawImage(videoElement, 0, 0, videoCanvas.width, videoCanvas.height); drawDetection(tempCtx, data.detections, data.image_shape.width, data.image_shape.height, videoCanvas.width, videoCanvas.height); } }) .catch(error => { console.error('Frame detection error:', error); }); } // --- 事件监听器 --- // 更新API端点 updateApiBtn.addEventListener('click', () => { const newUrl = apiUrlInput.value.trim(); if (newUrl) { apiEndpoint = newUrl; updateStatus(`API端点已更新为: ${apiEndpoint}`, false, true); document.querySelector('footer code').textContent = newUrl.split('/detect')[0] || newUrl; } }); // 图片文件选择 fileInput.addEventListener('change', function(e) { const file = e.target.files[0]; if (!file) return; const reader = new FileReader(); reader.onload = function(event) { const img = new Image(); img.onload = function() { // 可选:在检测前预览图片到结果画布 resultCanvas.width = img.width; resultCanvas.height = img.height; resultCtx.drawImage(img, 0, 0); updateStatus(`已加载图片: ${file.name} (${img.width}x${img.height})`); }; img.src = event.target.result; // 存储DataURL,供检测按钮使用 resultCanvas.dataset.currentImage = event.target.result; }; reader.readAsDataURL(file); }); // 检测图片按钮 detectImageBtn.addEventListener('click', function() { const currentImage = resultCanvas.dataset.currentImage; if (!currentImage) { updateStatus('请先选择一张图片。', true); return; } handleImageDetection(currentImage); }); // 清空图片画布 clearImageBtn.addEventListener('click', function() { resultCtx.clearRect(0, 0, resultCanvas.width, resultCanvas.height); resultCanvas.removeAttribute('data-current-image'); fileInput.value = ''; updateStatus('画布已清空。'); }); // 开启摄像头 startCamBtn.addEventListener('click', function() { if (navigator.mediaDevices && navigator.mediaDevices.getUserMedia) { const constraints = { video: { facingMode: 'user' } }; // 使用前置摄像头 navigator.mediaDevices.getUserMedia(constraints) .then(function(stream) { videoStream = stream; videoElement.srcObject = stream; videoElement.style.display = 'block'; startCamBtn.disabled = true; stopCamBtn.disabled = false; updateStatus('摄像头已开启。点击“停止检测”以结束。'); // 视频元数据加载后开始检测 videoElement.onloadedmetadata = () => { startVideoDetection(); }; }) .catch(function(err) { updateStatus(`无法访问摄像头: ${err.name}`, true); console.error('Camera error:', err); }); } else { updateStatus('您的浏览器不支持摄像头API。', true); } }); // 停止摄像头检测 stopCamBtn.addEventListener('click', function() { stopVideoDetection(); if (videoStream) { videoStream.getTracks().forEach(track => track.stop()); videoStream = null; } videoElement.srcObject = null; videoElement.style.display = 'none'; videoCtx.clearRect(0, 0, videoCanvas.width, videoCanvas.height); startCamBtn.disabled = false; stopCamBtn.disabled = true; updateStatus('摄像头已关闭。'); }); // FPS滑块 fpsSlider.addEventListener('input', function() { currentFps = parseInt(this.value); fpsValue.textContent = currentFps; if (detectionInterval) { // 如果正在检测,则重启间隔以应用新的FPS stopVideoDetection(); startVideoDetection(); } }); // 初始化状态 updateStatus('Demo已加载。请选择图片或开启摄像头开始体验。'); });

JavaScript代码核心逻辑解析

  1. 状态管理:使用updateStatus函数统一更新页面底部的状态栏,提供操作反馈。
  2. 绘制函数drawDetection函数负责在画布上绘制检测框和标签。它处理了图像缩放和居中,确保检测框能正确对应到画布上的位置。
  3. 图片检测流程
    • handleImageDetection: 将图片的Base64数据发送到后端API,收到结果后,在resultCanvas上绘制原图和检测框。
  4. 视频检测流程
    • startVideoDetection/stopVideoDetection: 控制定时器的启动和停止,实现按固定频率(FPS)捕获帧并检测。
    • captureAndDetectFrame: 核心循环函数。它将当前视频帧绘制到videoCanvas,转换为Base64,发送给后端,并在收到结果后立即在同一个画布上绘制检测框,实现“实时”覆盖的效果。
  5. 事件绑定:将所有按钮、滑块、输入框的点击和变化事件与对应的函数绑定起来。
  6. API端点配置:允许用户在前端界面直接修改后端API的URL,提高了Demo的灵活性。

5. 前后端联调与效果展示

代码都写好了,现在是让它们“握手”并跑起来的时候了。

5.1 启动与连接

  1. 启动后端:确保你的Flask后端服务正在运行(python app.py)。终端应显示服务运行在http://0.0.0.0:5000
  2. 启动前端:由于前端使用了Fetch API,直接通过浏览器打开index.html文件(file://协议)可能会因跨域问题(CORS)导致请求失败。最简单的方式是使用一个简单的HTTP服务器来托管前端文件。
    • 在项目根目录下打开另一个终端。
    • 使用Python快速启动一个HTTP服务器:
      # Python 3 python -m http.server 8080
    • 或者使用Node.js的http-server(需提前安装:npm install -g http-server):
      http-server -p 8080
  3. 访问页面:打开浏览器,访问http://localhost:8080(或你指定的端口)。你应该能看到我们设计的界面。

5.2 功能测试

现在,让我们逐一测试所有功能:

  • 图片检测

    1. 在左侧“图片检测”面板,点击“选择文件”按钮,上传一张包含物体(如人、车、动物)的图片。
    2. 图片会预览在下方画布中。
    3. 点击“检测图片”按钮。观察状态栏,会显示“正在发送图片进行检测...”,稍后会变为“检测成功!共发现 X 个目标。”。
    4. 同时,画布上会在检测到的物体周围绘制出绿色的边界框和标签。
  • 实时视频检测

    1. 在右侧“实时视频检测”面板,点击“开启摄像头”按钮。浏览器会请求摄像头权限,请点击“允许”。
    2. 你的摄像头画面会显示在下方画布中。
    3. 检测会自动开始。你会看到画面上实时出现绿色的检测框,跟随物体移动。
    4. 拖动“检测频率”滑块,可以调整每秒发送多少帧去检测。调低可以减轻后端压力,调高可以获得更流畅的检测体验。
    5. 点击“停止检测”按钮,会停止检测并关闭摄像头。
  • 切换后端API:如果你将后端部署到了另一台服务器(例如云服务器),你可以在页面底部的输入框中修改API地址(例如http://your-server-ip:5000/detect),然后点击“更新”按钮。之后的所有检测请求都会发送到新的地址。

5.3 效果展示与体验

当这一切顺利运行起来,你会获得一个功能完整的交互式目标检测Web应用。它的价值在于:

  1. 直观性:用户无需理解命令行或Python脚本,通过直观的网页界面即可操作。
  2. 实时性:视频检测功能提供了近乎实时的反馈,让模型的动态检测能力一目了然。
  3. 可交互性:用户可以自由切换图片/视频模式,调整检测频率,体验不同的交互。
  4. 可移植性:前端是纯静态文件(HTML, JS, CSS),可以轻松部署到任何Web服务器或对象存储。后端API也可以独立部署和扩展。

这个Demo本身已经是一个可用的产品原型。你可以在此基础上,继续添加更多功能,比如:

  • 选择不同的模型(例如切换DAMOYOLO的不同尺寸版本)。
  • 增加检测结果的统计信息面板。
  • 允许用户下载带检测框的图片。
  • 添加更多预处理或后处理选项(如置信度阈值调整)。
  • 使用WebSocket替代HTTP轮询,实现更低延迟的视频流检测。

6. 总结

通过这个完整的项目,我们实践了将AI模型能力产品化的关键一步:构建一个用户友好的交互界面。我们从零开始,搭建了一个基于Flask的轻量级检测API后端,并用原生JavaScript开发了一个功能丰富的前端应用,实现了图片上传检测和实时视频流检测。

整个过程的核心思路非常清晰:前端负责采集和展示,后端负责计算和返回,JSON作为沟通的语言。这种解耦的架构使得前后端可以独立开发和优化。即使你之后想用更强大的前端框架(如Vue、React)重构界面,或者用性能更高的后端框架(如FastAPI)重写服务,整体的通信协议和业务流程都无需改变。

对于初学者来说,这个项目涵盖了现代Web AI应用的基础要素:RESTful API设计、异步网络请求(Fetch)、Canvas绘图、媒体设备访问(getUserMedia)以及基础的UI/UX设计。它不仅仅是一个Demo,更是一个可以延伸和扩展的模板。

将AI模型从实验室的Jupyter Notebook搬到真实的Web环境中,让其能够被更多人直接使用和体验,是AI技术产生实际价值的重要一环。希望这个结合了DAMOYOLO-S与JavaScript的实践,能为你打开一扇门,启发你创造出更多有趣、有用的AI应用。


获取更多AI镜像

想探索更多AI镜像和应用场景?访问 CSDN星图镜像广场,提供丰富的预置镜像,覆盖大模型推理、图像生成、视频生成、模型微调等多个领域,支持一键部署。

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询