README.md

metadata

title: YOLO图像分割应用
emoji: 🔍
colorFrom: blue
colorTo: purple
sdk: gradio
sdk_version: 5.12.0
app_file: app.py
pinned: false

Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

YOLO 图像分割应用

这是一个基于 YOLOv8 的图像分割应用，可以对上传的图片进行实例分割。应用会自动检测和分割图像中的对象，并为每个类别生成单独的可视化结果。

功能特点

• 支持多类别实例分割
• 可调节的参数：
  • 置信度阈值 (0.1-1.0)
  • IOU阈值 (0.1-1.0)
  • 掩码阈值 (0.1-1.0)
  • 线条粗细 (1-5)
• 支持高分辨率掩码选项
• 自动颜色编码不同类别
• 实时可视化分割结果

使用方法

1. 上传一张图片
2. 根据需要调整参数：
   • 置信度阈值：降低可以检测更多对象，提高可以减少误报
   • IOU阈值：控制重叠检测的过滤程度
   • 掩码阈值：控制分割掩码的精确度
   • 线条粗细：调整边界框的视觉效果
3. 选择是否启用高分辨率掩码
4. 点击"开始分割"按钮
5. 查看分割结果：
   • 完整结果：显示所有检测到的对象
   • 单类别结果：每个检测到的类别单独显示

技术栈

• Ultralytics YOLO v8.0.196
• Gradio 4.44.1
• OpenCV-Headless 4.8.1
• NumPy 1.24.3
• PyTorch 2.0+

注意事项

• 首次运行时会自动下载预训练模型
• 处理大尺寸图片时可能需要较长时间
• 启用高分辨率掩码会增加处理时间但提供更精确的分割结果
• 本应用使用无头版本的 OpenCV，优化了服务器端运行性能