一、开源项目简介

FuturoGen - YOLOv8 Web Platform

一个基于Python Flask的Web应用，用于YOLOv8目标检测模型的图片标注、数据集管理、模型训练和模型导出，由Boyan开发。

二、开源协议

使用Apache-2.0开源协议

三、界面展示

运行截图

四、功能概述

FuturoGen - YOLOv8 Web Platform是一个基于YOLOv8的目标检测Web平台，提供从数据管理、图像标注、模型训练到模型导出的一站式解决方案。

功能特性

1. 项目管理
   创建、列出、删除项目
2. 图片管理

• 上传单张或多张图片
• 通过ZIP文件批量上传图片
• 拖拽上传图片

3. 图片标注

• 基于Web的边界框标注功能
• 图片间导航（上一张/下一张）
• 项目内标签（类别）管理：添加、编辑、删除
• 标注数据实时保存到数据库，并自动生成YOLOv8 .txt 格式的标注文件

4. 数据集划分
   批量将图片标记为训练集、验证集、测试集或未分配
5. 模型训练

• 实际调用 ultralytics 库进行模型训练
• 训练任务在后台线程中运行，前端实时显示训练日志和进度
• 自动准备训练所需的数据集目录结构和 data.yaml 文件

6. 模型导出
   将训练好的模型导出为ONNX、TorchScript等多种格式

五、技术选型

技术栈

• 后端
  Python 3.9+, Flask, Flask-SQLAlchemy, threading, OpenCV (cv2), Pillow (PIL.Image), ultralytics, python-dotenv, PyYAML
• 前端
  HTML5, CSS3, JavaScript, Bootstrap 5, Jinja2, Fabric.js
• 数据库
  SQLite
• 部署
  直接运行 python app.py

项目结构


yolov8_web_platform/
├── app.py                 # Flask应用入口
├── requirements.txt       # 项目依赖
├── .env                   # 环境变量配置
├── models.py             # 数据库模型
├── routes.py             # 路由处理
├── README.md             # 项目说明文档
├── static/               # 静态文件目录
│   ├── uploads/          # 上传图片存储目录
│   ├── datasets/         # 数据集目录
│   └── models/           # 模型存储目录
└── templates/            # HTML模板目录
    ├── base.html         # 基础模板
    ├── index.html        # 首页
    ├── project_detail.html # 项目详情页
    ├── images.html       # 图片管理页
    ├── annotate.html     # 图片标注页
    ├── dataset.html      # 数据集划分页
    ├── train.html        # 模型训练页
    └── export.html       # 模型导出页

安装与运行

pip install -r requirements.txt

在浏览器中访问 http://localhost:5005