# 视频数据压缩与快速检索系统

**Repository Path**: scp115/video_compression_and_fast_search_system

## Basic Information

- **Project Name**: 视频数据压缩与快速检索系统
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 15
- **Created**: 2025-12-16
- **Last Updated**: 2025-12-16

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 视频压缩与快速搜索系统

这是一个用于视频压缩和快速搜索的系统，主要使用Python实现。系统包含视频摘要生成、关键帧提取、特征提取、摘要生成、摘要评估、数据集生成、模型训练、搜索服务等多个模块。

## 项目结构

- **Video-Summarization-Pytorch**: 视频摘要生成模块，包含数据集生成、特征提取、模型定义、训练和测试代码。
- **milvus**: 向量数据库模块，用于存储和快速检索视频特征向量。
- **appNew.py**: 主界面程序，提供视频选择、摘要生成、关键帧展示等功能。
- **SearchServer.py**: 搜索服务程序，提供基于特征的视频搜索功能。
- **func_extra_frame.py**: 关键帧提取工具模块。
- **extra_frame.py**: 关键帧展示界面模块。

## 功能模块说明

### 视频摘要生成

使用深度学习模型对视频进行摘要生成，提取视频中的关键帧，减少视频冗余信息。

- **Video-Summarization-Pytorch/networks/DSN.py**: 摘要生成网络。
- **Video-Summarization-Pytorch/utils/vsum_tool.py**: 摘要评估工具。
- **Video-Summarization-Pytorch/video_summary.py**: 摘要生成主程序。

### 关键帧提取

从视频中提取关键帧，用于后续的特征提取和搜索。

- **func_extra_frame.py**: 提供关键帧提取算法。
- **extra_frame.py**: 提供关键帧展示界面。

### 特征提取

使用CNN网络提取视频帧的特征向量，用于向量数据库存储和相似性搜索。

- **Video-Summarization-Pytorch/networks/CNN.py**: CNN特征提取网络。
- **Video-Summarization-Pytorch/utils/generate_dataset.py**: 用于生成特征数据集。

### 模型训练

提供训练视频摘要模型的功能。

- **Video-Summarization-Pytorch/train_video_summarisation.py**: 模型训练主程序。
- **Video-Summarization-Pytorch/model**: 存放训练好的模型文件。

### 向量数据库

使用Milvus向量数据库存储视频特征向量，实现快速检索。

- **milvus**: Milvus数据库配置和数据存储目录。
- **milvus/docker-compose.yml**: Milvus数据库启动配置。

### 搜索服务

提供基于特征向量的视频搜索功能。

- **SearchServer.py**: 搜索服务主程序。

### 图形界面

提供用户交互界面，支持视频选择、摘要生成、关键帧展示等功能。

- **appNew.py**: 主界面程序。
- **app.ui**: 主界面布局文件。
- **extra_frame.ui**: 关键帧展示界面布局文件。

## 安装与运行

### 依赖安装

```bash
pip install torch torchvision numpy scikit-learn
```

### Milvus数据库启动

```bash
cd milvus
docker-compose up -d
```

### 模型训练

```bash
cd Video-Summarization-Pytorch
python train_video_summarisation.py
```

### 视频摘要生成

```bash
cd Video-Summarization-Pytorch
python video_summary.py
```

### 启动搜索服务

```bash
python SearchServer.py
```

### 启动图形界面

```bash
python appNew.py
```

## 使用说明

1. 打开`appNew.py`启动图形界面。
2. 点击“选择视频”按钮，选择需要处理的视频文件。
3. 点击“开始处理”按钮，系统将自动提取关键帧并生成视频摘要。
4. 使用`SearchServer.py`进行视频内容搜索。

## 许可证

本项目使用MIT许可证。详见[MIT License](https://opensource.org/licenses/MIT)。