# onnx-pose

**Repository Path**: sulv0302/onnx-pose

## Basic Information

- **Project Name**: onnx-pose
- **Description**: 基于java/onnx实现的姿态识别，支持图片，视频，流地址播放查看姿态识别
- **Primary Language**: Unknown
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 20
- **Forks**: 3
- **Created**: 2025-04-11
- **Last Updated**: 2025-06-24

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# onnx-inference4j-play

### 🎯 项目简介

`pose-onnx` 是一个基于 **Java** 和 **ONNX Runtime** 的姿态识别工具✨

### 🏗️ 软件架构

- **Java 17**
- **CUDA 11.8** (可以根据需要在 `onnxruntime-gpu` 依赖中适配不同版本)
- 基于 **ONNX Runtime** 和 **OpenCV** 实现图像与视频处理
- 采用 **FlatLaf** 提供现代化的 Swing UI 风格

### 📦 依赖环境

#### **主要依赖**

- **onnxruntime-gpu**: 提供 ONNX 模型的 GPU 推理支持
- **flatlaf**: 用于美化 Swing 界面，提供现代风格
- **opencv**: 负责图像和视频处理操作
- **lombok**: 简化 Java 代码的 Getter/Setter 等样板代码
- **fastjson**: 用于高效的 JSON 序列化与反序列化

### 💻 安装步骤

1. **安装 Java 17**
   项目基于 **Java 17**，请确保系统已安装该版本。
   下载地址：[Java SE Development Kit 17](https://www.oracle.com/java/technologies/javase-jdk8-downloads.html)

2. **安装 CUDA 11.8**
   如果使用 GPU 推理功能，请确保安装 CUDA 11.8，或根据你的 GPU 环境调整 `onnxruntime-gpu` 依赖版本。

3. **克隆代码并构建**
   使用以下命令克隆并构建项目：

   ```
   bash复制代码git clone https://gitee.com/sulv0302/onnx-inference4j-play.git
   cd onnx-inference4j-play
   mvn clean install
   复制项目中的lib目录到打包后的jar包同层级
   ```

### 📝 使用说明

1. **拖拽模型**到模型区域
2. **选择视频、图片**或填写**流地址**
3. 点击**播放按钮**即可开始推理识别

功能特性：

- **图像识别**、**视频识别** 和 **流地址识别**
- **实时切换** 图片、视频源，支持 **拖拽文件**

### 🛠 功能概览

- **拖拽文件支持**：用户可以通过拖拽模型、图片、视频文件到界面进行加载
- **模型加载**：支持 ONNX 格式模型，配合标签文件进行推理
- **视频/图像推理**：可加载本地视频、图片或通过流地址进行实时推理

#### 选择视频或图片

点击“选择视频”或“选择图片”按钮，选择文件后即可进行推理，支持的视频格式包括 **mp4**、**avi**、**mkv** 等。


### 🙌 参与贡献

1. Fork 本仓库
2. 新建 Feat_xxx 分支
3. 提交代码
4. 新建 Pull Request

欢迎提交 bug 报告和建议。


![输入图片说明](src/main/resources/img.png)

![输入图片说明](4973b1e0883073501882da62b2e15028.jpg)