# MLLM_ImgSorter

**Repository Path**: csbobo/MLLM_ImgSorter

## Basic Information

- **Project Name**: MLLM_ImgSorter
- **Description**: 图片分类查找工具,支持图片内容分析、人脸识别、相似图片搜索等功能
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-10-18
- **Last Updated**: 2024-11-05

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 图片管理与搜索工具

这是一个本地运行的图片分类查找工具,支持图片内容分析、人脸识别、相似图片搜索等功能。

## 功能特点

支持通过输入描述、上传图片、选择类型、选择人脸等方式进行搜索。搜索结果可以导出至指定文件夹。

## 使用技术

- 使用face_recognition库进行人脸识别
- 使用多模态模型进行图片内容分析
- 使用向量模型进行图片描述比对
- 使用支持OpenAI兼容API，默认使用的是ollama的兼容API
- 使用本地运行ollama让所有功能在本地运行，不依赖云服务，安全高效

## 系统要求

- Qt 6.0或更高版本
- AI服务支持(支持OpenAI兼容API)，默认使用的是ollama的兼容API

## 配置说明

### AI服务设置

系统需要配置两个AI服务：
1. Chat AI服务：用于图片内容分析(推荐使用minicpm-v)
2. Embedding AI服务：用于生成内容向量(推荐使用bge-m3)

### 路径设置

需要配置两个主要路径：
1. 已知人脸文件夹：存放用于人脸识别的参考图片(默认路径为程序所在目录的face文件夹)
2. 扫描文件夹：需要管理的图片文件所在文件夹(默认路径为程序所在目录的fileDir文件夹)

已知人脸文件夹中需要根据人脸图片的名称来命名人脸，例如face/张三.jpg，则张三就是人脸名称。

## 使用说明

### 1. 基础设置
- 点击"修改文件路径"设置必要的文件夹路径
- 点击"AI服务设置"配置AI服务参数

**注意：**
- 使用前确保已知人脸文件夹中有人脸图片，否则无法进行人脸搜索
- 使用前确保AI服务正常运行，可以点击测试按钮测试AI服务是否正常

### 2. 扫描与更新
- 使用"全部重新扫描"进行完整扫描
- 使用"新增更新扫描"进行增量更新
- 可以单独更新人脸识别、内容类型、描述等信息

#### 扫描操作步骤说明：

**在没有已知人脸图片的情况下，扫描新文件夹操作步骤:**
1. 点击"重新扫描文件" 
2. 点击"重新扫描人脸"
3. 在已知人脸文件夹中修改文件名称，用于命名人脸，删除非人脸的图片
4. 点击"重新识别人脸"
5. 点击"重新生成类型"
6. 点击"重新生成描述"
7. 点击"重新生成嵌入向量"

**在已知人脸图片的情况下，扫描新文件夹操作步骤:**
1. 点击"全部重新扫描"
2. 等待扫描完成，扫描结果会显示在界面中

> "全部重新扫描"相当于依次点击"重新扫描文件"、"重新识别人脸"、"重新生成类型"、"重新生成描述"、"重新生成嵌入向量"

**新增图片更新：**
- 在待扫描文件夹中新增图片后，点击"新增更新扫描"，则新增的图片会进行扫描，并更新到已有的信息中

### 3. 搜索功能
- 内容搜索：输入描述或上传图片进行搜索
- 类型搜索：按预定义类型搜索
- 人脸搜索：按人名搜索
- 关键词搜索：在内容描述中搜索关键词

## 注意事项

1. 首次使用需要正确配置AI服务和文件路径
2. 大量图片的扫描和分析可能需要较长时间
3. 建议定期备份file_info.json文件

## 贡献

欢迎提交Issue和Pull Request。