UE5 Plugin NNEngine - Neural Network Engine

[Pumpkin]

V5.0

Пожалуйста, авторизуйтесь для просмотра ссылки.

Трейлер / Trailer:

Простой, ускоренный вывод машинного обучения из BP и C++ с использованием собственной библиотеки ONNX Runtime.
Демонстрационное видео: Обзор, Демонстрация монокулярной оценки глубины, Демонстрация переноса художественного стиля
Учебное видео: реализация оценки глубины

Документация: ссылка
Просто вызвав несколько узлов Blueprint, вы можете загрузить и запустить передовой ИИ.
Этот плагин поддерживает ONNX (Open Neural Network Exchange), широко используемый формат модели машинного обучения с открытым исходным кодом.
Многие фреймворки машинного обучения, такие как PyTorch и TensorFlow, могут экспортировать свою модель в формат ONNX.

Многие обученные модели доступны в зоопарке моделей ONNX.
Производительность является нашим первым соображением.
Этот плагин поддерживает оптимизацию модели во время выполнения и ускорение графического процессора на различном оборудовании, а также сам этот плагин оптимизирован.
Демонстрационный проект содержит практические примеры

Обнаружение человека
Оценка позы человека
Распознавание лиц
Оценка ориентиров лица
Отслеживание взгляда с помощью одной RGB-камеры.

Также примеры проектов для

Оценка глубины с помощью монокулярной RGB-камеры
Произвольная передача художественного стиля доступны.

Технические детали / Technical Details:

Features:

• Loads ONNX models at runtime. Automatically optimizes the model when loaded.
• Runs ONNX models.
• (On Windows) Supports hardware acceleration by DirectML on DirectX 12 capable GPUs.
• (On Windows and Ubuntu) Supports hardware acceleration by CUDA and TensorRT on supported NVIDIA GPUs.
• (On Windows) Gets a list of GPU names available on the system.
• Processes (resize, crop, rotate) UTexture and converts it to int8 array.

Code Modules:

• OnnxRuntime (Runtime)
• TextureProcessing (Runtime)
• CustomizedOpenCV (Runtime)
• DirectXUtility (Runtime)

Number of Blueprints: 3
Number of C++ Classes: 7+
Network Replicated: No
Supported Development Platforms: Windows 10 64bit, Ubuntu 18.04.6 Desktop 64bit
Supported Target Build Platforms: Windows 10 64bit, Ubuntu 18.04.6 Desktop 64bit
Documentation:

Пожалуйста, авторизуйтесь для просмотра ссылки.

Example Project:

1. Пожалуйста, авторизуйтесь для просмотра ссылки.
2. Пожалуйста, авторизуйтесь для просмотра ссылки.
3. Пожалуйста, авторизуйтесь для просмотра ссылки.

Important/Additional Notes:

• Demo Project 1 is distributed as a C++ project. You need to install Visual Studio to compile it.