mirror of
https://github.com/wwhai/generic-rtsp-yolov8-render.git
synced 2026-04-22 15:17:26 +08:00
dev: update
This commit is contained in:
wwhai
@@ -40,12 +40,6 @@ void NV12ToRGB(uint8_t *y_plane, uint8_t *uv_plane, int width,
|
||||
/// @param frame
|
||||
void SDLDisplayNV12Frame(SDL_Renderer *renderer, SDL_Texture *texture, AVFrame *frame);
|
||||
|
||||
/// @brief
|
||||
/// @param renderer
|
||||
/// @param texture
|
||||
/// @param rect
|
||||
void SDLDrawRect(SDL_Renderer *renderer, SDL_Texture *texture, SDL_Rect *rect);
|
||||
|
||||
/// @brief
|
||||
/// @param renderer
|
||||
/// @param texture
|
||||
|
||||
@@ -1,25 +1,9 @@
|
||||
<!--
|
||||
Copyright (C) 2025 wwhai
|
||||
### 仓库介绍
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU Affero General Public License as
|
||||
published by the Free Software Foundation, either version 3 of the
|
||||
License, or (at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU Affero General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU Affero General Public License
|
||||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
-->
|
||||
# RTSP Monitor
|
||||
|
||||
## 项目简介
|
||||
RTSP Monitor 是一个基于 **libav*** 和 **SDL2** 的轻量级监控软件，支持从 RTSP 流拉取视频并渲染到图形界面，同时对每一帧进行图像处理。
|
||||
从 RTSP 流拉取视频并渲染到图形界面，同时对每一帧进行图像处理。
|
||||
|
||||
### 功能特性
|
||||
|
||||
1. **RTSP 拉流**：通过提供 RTSP URL，从流媒体服务器拉取视频流。
|
||||
2. **多线程处理**：
|
||||
- 线程 A：负责实时播放原始视频帧。
|
||||
@@ -28,9 +12,8 @@ RTSP Monitor 是一个基于 **libav*** 和 **SDL2** 的轻量级监控软件，
|
||||
- 矩形框为红色，标记文本为绿色。
|
||||
4. **硬件加速支持**：SDL2 硬件加速提升渲染性能。
|
||||
|
||||
---
|
||||
### 项目结构
|
||||
|
||||
## 项目结构
|
||||
```
|
||||
.
|
||||
├── include/ # 头文件目录
|
||||
@@ -49,9 +32,8 @@ RTSP Monitor 是一个基于 **libav*** 和 **SDL2** 的轻量级监控软件，
|
||||
├── README.md # 项目说明文档
|
||||
```
|
||||
|
||||
---
|
||||
### 环境要求
|
||||
|
||||
## 环境要求
|
||||
- **开发语言**：C
|
||||
- **依赖库**：
|
||||
- libavformat
|
||||
@@ -61,74 +43,45 @@ RTSP Monitor 是一个基于 **libav*** 和 **SDL2** 的轻量级监控软件，
|
||||
- SDL2
|
||||
- **系统环境**：支持 Linux、Windows 等平台
|
||||
|
||||
---
|
||||
### 编译与运行
|
||||
|
||||
#### 编译项目
|
||||
|
||||
## 编译与运行
|
||||
### 编译项目
|
||||
确保系统已安装依赖库，并通过 `pkg-config` 正确配置库路径。执行以下命令编译项目：
|
||||
|
||||
```bash
|
||||
make
|
||||
```
|
||||
|
||||
### 清理文件
|
||||
#### 清理文件
|
||||
|
||||
删除生成的中间文件和可执行文件：
|
||||
|
||||
```bash
|
||||
make clean
|
||||
```
|
||||
|
||||
### 运行程序
|
||||
#### 运行程序
|
||||
|
||||
编译成功后，运行以下命令启动程序：
|
||||
|
||||
```bash
|
||||
./rtsp_monitor <RTSP_URL>
|
||||
```
|
||||
|
||||
示例：
|
||||
|
||||
```bash
|
||||
./rtsp_monitor rtsp://192.168.1.10:554/live
|
||||
```
|
||||
|
||||
---
|
||||
### 核心模块说明
|
||||
|
||||
## 核心模块说明
|
||||
### 1. RTSP 拉流模块 (`rtsp_client.c`)
|
||||
负责通过 RTSP URL 拉取视频流，解析为帧并推送到帧队列。
|
||||
1. **RTSP 拉流模块 (`rtsp_client.c`)**
|
||||
负责通过 RTSP URL 拉取视频流，解析为帧并推送到帧队列。
|
||||
|
||||
### 2. 视频渲染模块 (`video_renderer.c`)
|
||||
负责从队列中读取原始视频帧，并通过 SDL2 硬件加速渲染到图形界面。
|
||||
2. **视频渲染模块 (`video_renderer.c`)**
|
||||
负责从队列中读取原始视频帧，并通过 SDL2 硬件加速渲染到图形界面。
|
||||
|
||||
### 3. 视频处理模块 (`video_processor.c`)
|
||||
负责对帧进行识别处理（具体识别逻辑由用户实现），并生成矩形框及标记信息。
|
||||
|
||||
### 4. 帧队列模块 (`frame_queue.c`)
|
||||
实现一个线程安全的队列，用于存储从 RTSP 流中拉取的视频帧。
|
||||
|
||||
---
|
||||
|
||||
## 注意事项
|
||||
1. **硬件加速**：
|
||||
- 确保系统支持 SDL2 的硬件加速功能，并正确启用相关依赖。
|
||||
2. **依赖库安装**：
|
||||
- 在 Linux 上安装依赖库：
|
||||
```bash
|
||||
sudo apt install libsdl2-dev libavcodec-dev libavformat-dev libavutil-dev libswscale-dev
|
||||
```
|
||||
3. **多线程安全**：
|
||||
- 本项目中的帧队列模块使用 `pthread` 实现线程安全，确保多线程访问不冲突。
|
||||
|
||||
---
|
||||
|
||||
## TODO
|
||||
1. 优化队列内存管理，支持动态扩容。
|
||||
2. 增加支持多种流媒体协议（如 HTTP、HLS）。
|
||||
3. 提供更多检测结果的展示方式。
|
||||
|
||||
---
|
||||
|
||||
## 许可证
|
||||
本项目遵循 [MIT License](LICENSE) 协议开源。
|
||||
|
||||
---
|
||||
|
||||
## 联系方式
|
||||
如果有任何问题或建议，请联系作者：
|
||||
- **邮箱**：cnwwhai@gmail.com
|
||||
3. **视频处理模块 (`video_processor.c`)**
|
||||
负责对帧进行识别处理（具体识别逻辑由用户实现），并生成矩形框及标记信息。
|
||||
|
||||
+3
-12
@@ -88,16 +88,7 @@ void SDLDisplayNV12Frame(SDL_Renderer *renderer, SDL_Texture *texture, AVFrame *
|
||||
SDL_RenderCopy(renderer, texture, NULL, NULL);
|
||||
free(rgb_buffer);
|
||||
}
|
||||
// 绘制矩形
|
||||
// @param renderer 渲染器
|
||||
// @param texture 纹理
|
||||
// @param rect 矩形
|
||||
void SDLDrawRect(SDL_Renderer *renderer, SDL_Texture *texture, SDL_Rect *rect)
|
||||
{
|
||||
SDL_SetRenderDrawColor(renderer, 255, 0, 0, 255);
|
||||
SDL_RenderDrawRect(renderer, rect);
|
||||
SDL_RenderPresent(renderer);
|
||||
}
|
||||
|
||||
// 绘制文本
|
||||
// @param renderer 渲染器
|
||||
// @param texture 纹理
|
||||
@@ -187,13 +178,13 @@ void SDLDrawBox(SDL_Renderer *renderer, TTF_Font *font, const char *text,
|
||||
// Render text above the rectangle
|
||||
if (text != NULL)
|
||||
{
|
||||
SDL_Color textColor = {255, 0, 0, 0}; // White color
|
||||
SDL_Color textColor = {255, 0, 0, 0};
|
||||
SDL_Surface *textSurface = TTF_RenderText_Solid(font, text, textColor);
|
||||
SDL_Texture *textTexture = SDL_CreateTextureFromSurface(renderer, textSurface);
|
||||
|
||||
SDL_Rect textRect;
|
||||
textRect.x = x;
|
||||
textRect.y = y - textSurface->h; // Position text above the rectangle
|
||||
textRect.y = y - 24;
|
||||
textRect.w = textSurface->w;
|
||||
textRect.h = textSurface->h;
|
||||
|
||||
|
||||
@@ -60,10 +60,10 @@ void *video_renderer_thread(void *arg)
|
||||
SDL_DestroyRenderer(renderer);
|
||||
SDL_DestroyWindow(window);
|
||||
SDL_Quit();
|
||||
return NULL;
|
||||
pthread_exit(NULL);
|
||||
}
|
||||
// 加载字体
|
||||
TTF_Font *font = TTF_OpenFont("SourceHanMonoSC-Light.ttf", 24);
|
||||
TTF_Font *font = TTF_OpenFont("SourceHanMonoSC-Light.ttf", 18);
|
||||
if (font == NULL)
|
||||
{
|
||||
fprintf(stderr, "Failed to load font! TTF_Error: %s\n", TTF_GetError());
|
||||
@@ -123,8 +123,9 @@ void *video_renderer_thread(void *arg)
|
||||
{
|
||||
for (int i = 0; i < boxes_item.box_count; ++i)
|
||||
{
|
||||
RenderBox(renderer, &boxes_item.Boxes[i]);
|
||||
SDLDrawText(renderer, texture, font, boxes_item.Boxes[i].label, boxes_item.Boxes[i].x, boxes_item.Boxes[i].y);
|
||||
SDLDrawBox(renderer, font, boxes_item.Boxes[i].label,
|
||||
boxes_item.Boxes[i].x, boxes_item.Boxes[i].y,
|
||||
boxes_item.Boxes[i].w, boxes_item.Boxes[i].h, 1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user