基于java实现的onnx模型推理识别可视化窗口
一个基于 Java 和 ONNX Runtime 的推理识别工具,提供了直观的可视化窗口,支持多种场景下的模型推理,包括图像、视频、流媒体等。
它具备 多模型推理、目标跟踪、动态源切换 和 文件拖拽 功能,适用于各类推理场景,使用体验流畅友好。
软件架构- Java 8
- CUDA 11.8 (可以根据需要在 onnxruntime-gpu 依赖中适配不同版本)
- 基于 ONNX Runtime 和 OpenCV 实现图像与视频处理
- 采用 FlatLaf 提供现代化的 Swing UI 风格
依赖环境主要依赖- onnxruntime-gpu: 提供 ONNX 模型的 GPU 推理支持
- flatlaf: 用于美化 Swing 界面,提供现代风格
- opencv: 负责图像和视频处理操作
- lombok: 简化 Java 代码的 Getter/Setter 等样板代码
- fastjson: 用于高效的 JSON 序列化与反序列化
安装步骤使用说明- 拖拽模型到模型区域
- 选择视频、图片或填写流地址
- 点击播放按钮即可开始推理识别
功能特性: - 支持 多模型 推理识别
- 图像识别、视频识别 和 流地址识别
- 动态启用或禁用目标跟踪
- 实时切换 图片、视频源,支持 拖拽文件
功能概览- 拖拽文件支持:用户可以通过拖拽模型、图片、视频文件到界面进行加载
- 模型加载:支持 ONNX 格式模型,配合标签文件进行推理
- 视频/图像推理:可加载本地视频、图片或通过流地址进行实时推理
- 目标跟踪:提供简单的目标跟踪功能,可根据用户需要启用/禁用
使用示例加载模型用户可以从本地拖拽 .onnx 模型文件及对应的 .txt 标签文件至界面,或通过“选择模型”按钮加载。 选择视频或图片点击“选择视频”或“选择图片”按钮,选择文件后即可进行推理,支持的视频格式包括 mp4、avi、mkv 等。 启用目标跟踪在顶部的 “启用目标跟踪” 选项中,勾选或取消勾选以动态启用/禁用目标跟踪功能。
以下内容需要积分高于 2 才可浏览
链接: https://pan.baidu.com/s/1FHoUcLD1LkDE6oBPEo7FXg
提取码下载:
|