FastAPI YOLOv8 Detection Server

Мощный и масштабируемый сервер на FastAPI для обнаружения объектов с использованием моделей YOLOv8 и возможностями анализа размеров. Сервер предоставляет RESTful API для обработки изображений и видео, включая продвинутые функции оценки реальных размеров объектов с использованием калибровочных объектов.

🌟 Возможности

Обнаружение объектов: Поиск объектов на изображениях и видео с помощью моделей YOLOv8.
Анализ размеров: Оценка реальных размеров объектов с использованием калибровочных объектов (например, теннисных мячей).
Поддержка нескольких моделей: Возможность одновременной загрузки и использования нескольких моделей YOLOv8.
Асинхронная обработка: Неблокирующая система очереди задач для эффективной обработки медиафайлов.
Управление файлами: Загрузка моделей и медиафайлов через API-эндпоинты.
Поддержка Docker: Контейнеризированное развертывание с помощью Docker и Docker Compose.
Подача статических файлов: Прямой доступ к обработанным изображениям и результатам.
Подробное логирование: Детальное ведение логов для отладки и мониторинга.

🏗️ Архитектура

Компоненты системы

┌─────────────────┐    ┌──────────────────┐    ┌──────────────────┐
│   FastAPI App   │◄──►│  Task Manager    │◄──►│  Processors      │
└─────────────────┘    └──────────────────┘    └──────────────────┘
       │                        │                       │
       ▼                        ▼                       ▼
┌─────────────────┐    ┌──────────────────┐    ┌──────────────────┐
│    Routers      │    │   Task Queue     │    │  YOLO Models     │
└─────────────────┘    └──────────────────┘    └──────────────────┘
       │                        │                       │
       ▼                        ▼                       ▼
┌─────────────────┐    ┌──────────────────┐    ┌──────────────────┐
│   Services      │    │  Storage System  │    │  Model Loader     │
└─────────────────┘    └──────────────────┘    └──────────────────┘

Основные модули

Основное приложение (app/deepseek_one_file_todo.py)
Точка входа FastAPI.
Система управления задачами.
Регистрация API-маршрутов.
Управление задачами (app/our_task_manager.py)
Асинхронная очередь задач.
Очереди обработки фото и видео.
Отслеживание статуса задач.
Процессоры (app/processors/)
Photo Processor: Обнаружение объектов на изображениях.
Video Processor: Обнаружение объектов на видео.
Photo Size Processor: Анализ размеров с калибровкой.
Base Processor: Общая функциональность.
Роутеры (app/routers/)
Server Utility: Утилиты и управление файлами.
YOLO Processing: Эндпоинты обработки объектов.
Object Size: Эндпоинты анализа размеров.
Сервисы (app/services/)
Реализация бизнес-логики.
Операции с моделями и файлами.
Система хранения (storage/)
Requests: Входящие медиафайлы.
Responses: Обработанные результаты.
Models: Файлы моделей YOLO.
Logs: Логи приложения.
Uploads: Загруженные пользователями файлы.

🛠️ Технологии

Основной стек

FastAPI: Высокопроизводительный веб-фреймворк.
YOLOv8: Современные модели обнаружения объектов.
Ultralytics: Официальная реализация YOLO.
OpenCV: Компьютерное зрение и обработка изображений.
Python 3.8+: Язык программирования.

Зависимости

ultralytics — реализация YOLOv8.
fastapi[all] — веб-фреймворк с расширениями.
uvicorn — ASGI сервер.
aiohttp — асинхронный HTTP-клиент.
pydantic — валидация данных.
python-dotenv — управление переменными окружения.
opencv-python — библиотека компьютерного зрения.

🚀 Быстрый старт

Использование Docker (рекомендуется)

# Клонируйте репозиторий
git clone <repository-url>
cd fastapi_yolo_server

# Создайте файл .env из примера
cp .env.example .env

# Запустите сервер
docker-compose up -d --build

# Сервер будет доступен по адресу http://localhost:8000

Ручная установка

# Клонируйте репозиторий
git clone <repository-url>
cd fastapi_yolo_server

# Создайте виртуальное окружение
python -m venv venv
source venv/bin/activate  # Для Windows: venv\Scripts\activate

# Установите зависимости
pip install -r requirements.txt

# Запустите сервер
python app/deepseek_one_file_todo.py

⚙️ Конфигурация

Сервер использует переменные окружения для настройки. Основные параметры в .env:
- HOST: IP-адрес сервера (по умолчанию 0.0.0.0).
- PORT: Порт сервера (по умолчанию 8000).
- RESPONSE_HOST: Базовый URL для ответов.
- ENABLE_CUSTOM_FRONTEND: Включить ли кастомный фронтенд (true/false).

📂 Структура папок

app/: Основной код приложения.
storage/: Хранилище для моделей, логов, загрузок и результатов.
requirements.txt: Список зависимостей.
Dockerfile: Инструкции для сборки образа.
docker-compose.yml: Конфигурация Docker Compose.

GitFlic Catalog