项目简介
这是一个基于深度学习的图文素材检索系统,可通过自然语言搜索和图像搜索查找本地图片和视频素材。用户能使用文字描述或上传图片进行搜索,系统会在本地文件库中匹配相关素材。项目运用深度学习模型进行特征提取和匹配,支持中文搜索,还有自动扫描和缓存功能以提升搜索效率。
项目的主要特性和功能
- 文字搜索:支持中文,用户输入关键词可查找相关图片和视频。
- 图像搜索:用户上传图片可查找相似的图片和视频片段。
- 自动扫描:自动扫描指定目录下的图片和视频,将信息存入数据库便于后续搜索。
- 缓存机制:采用缓存机制提高搜索效率,先查询缓存,无结果再搜索并写入缓存。
- 深度学习模型:采用CLIP模型进行特征提取和匹配,支持多种格式的图片和视频文件。
- 灵活的配置选项:提供丰富配置选项,可按需调整扫描路径、模型选择、设备选择等。
安装使用步骤
- 安装依赖:在项目目录下运行
pip install -r requirements.txt
命令安装所需依赖库。 - 配置环境:依据
config.py
文件说明,设置项目配置参数,如扫描路径、设备选择等。 - 启动程序:运行
main.py
文件启动程序,程序会自动扫描配置文件指定目录,并将图片和视频信息存入数据库。 - 进行搜索:通过API接口或网页前端搜索,输入关键词或上传图片,系统返回相关搜索结果。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】