Files
zydi-web/app/docs/function/pose.md
T
2025-01-23 09:05:45 +00:00

807 B

姿态检测模块 (pose.py)

功能描述

该模块使用YOLO模型进行人体姿态检测,并对检测到的人体区域进行裁剪保存。

主要组件

PoseMonitor类

处理姿态检测和图片裁剪的主要类。

主要方法

  • __init__(): 初始化YOLO模型和路径配置
  • _process_image(): 处理单张图片,检测人体并保存裁剪结果
  • monitor_directories(): 监控目录变化

工作流程

  1. 监控指定目录中的新图片
  2. 使用YOLO模型检测图片中的人体
  3. 对检测到的人体区域进行裁剪
  4. 将裁剪后的图片保存到指定目录

使用说明

  1. 确保YOLO模型路径配置正确
  2. 运行脚本开始监控图片目录
  3. 裁剪后的图片将保存在crop目录下

依赖

  • YOLO
  • PIL
  • OpenCV
  • Ultralytics