# retrieval-mf
retrieval-multi-feature 基于多特征融合的图像检索系统。
request:`python >= 3.9 torch>=1.10.1`
## Introduction
这是一个基于多特征融合的图像检索系统。核心部分主要分为**图像特征的提取**与**特征向量检索**两部分:
- 特征提取。目前主要包括传统特征:颜色、纹理、glcm特征;深度学习特征:vgg、vit。
- 向量检索。使用向量数据库milvus,具体见`docs/design.md`。
特征向量存放在milvus中便于比对,图片存储在文件服务器,其他元数据信息存储在mysql中。
用户首先需要将图片文件夹上传至服务器,然后开始提取特征形成特征库。之后用户可以通过上传一张图片来检索已经入库的文件中最相似的图片,并展示在页面上。
## Install
### milvus
milvus是专门设计用于处理对输入向量的查询的数据库,它能够以万亿级对向量进行索引。
使用`docker-compose`启动,具体见:[Install Milvus Standalone with Docker Compose Milvus documentation](https://milvus.io/docs/install_standalone-docker.md)
### attu
attu是milvus的开源可视化管理工具,不安装系统也可以正常启动。
docker安装启动attu,具体见:[Install Attu with Docker Compose Milvus documentation](https://milvus.io/docs/attu_install-docker.md)
```bash
docker run -p 8000:3000 -e MILVUS_URL={your machine IP}:19530 zilliz/attu:v2.2.3
```
### mysql
建议使用`docker`安装启动,如:
```bash
docker run --restart=always -itd --name mysql-test -p 3306:3306 -e MYSQL_ROOT_PASSWORD=123456 mysql
```
### fileserver
项目中实现了一个简单的HTTP文件服务器,在`utils/file_server.py`,进入`utils`文件夹后启动:
```bash
python file_server.py
```
### server
安装好python的虚拟环境后,安装依赖并启动项目:
```bash
# 安装依赖项
pip install -r requirements
# 启动项目,可以通过 -c 参数指定配置文件,默认为 conf/conf.ini
python server.py
```
## Usage
需要先建库,然后在库中搜索相似图片。
### 批量上传
确保需要添加的图片文件夹放在服务器可以访问到的位置!!!然后在批量上传界面输入文件夹的绝对路径,点击上传按钮,若成功,则可以看到后台正在处理的进度条。
![image](docs/pic/vue_adddir.png)
### 图像检索
上传一张图片,检索相似图片。其中可以通过下拉框选择不同的算法,通过调整截取框的位置使用部分图片进行检索。
需要注意,即使图片相似度很低,也依然会返回固定数量的结果。
![image](docs/pic/vue_search.png)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
<项目介绍> - 这是一个基于多特征融合的图像检索系统。核心部分主要分为图像特征的提取与特征向量检索两部分: 特征提取。目前主要包括传统特征:颜色、纹理、glcm特征;深度学习特征:vgg、vit。 向量检索。使用向量数据库milvus,具体见docs/design.md。 - 不懂运行,下载完可以私聊问,可远程教学 该资源内项目源码是个人的毕设,代码都测试ok,都是运行成功后才上传资源,答辩评审平均分达到96分,放心下载使用! 1、该资源内项目代码都经过测试运行成功,功能ok的情况下才上传的,请放心下载使用! 2、本项目适合计算机相关专业(如计科、人工智能、通信工程、自动化、电子信息等)的在校学生、老师或者企业员工下载学习,也适合小白学习进阶,当然也可作为毕设项目、课程设计、作业、项目初期立项演示等。 3、如果基础还行,也可在此代码基础上进行修改,以实现其他功能,也可用于毕设、课设、作业等。 下载后请首先打开README.md文件(如有),仅供学习参考, 切勿用于商业用途。 --------
资源推荐
资源详情
资源评论
收起资源包目录
基于多特征融合的图像检索系统.zip (69个子文件)
retrieval-mf-master
app
__init__.py 223B
app.py 614B
api
add.py 6KB
user.py 1KB
search.py 3KB
utils
enums.py 142B
task.py 1KB
scope.py 818B
error_type.py 2KB
jwt_verify.py 980B
route.py 646B
validators
base.py 663B
forms.py 3KB
LICENSE 10KB
tests
dao_user_test.py 2KB
1.jpg 112KB
dao_feature_test.py 1KB
cosine_test.py 477B
engine_test.py 922B
jwt_test.py 469B
milvus_test.py 738B
app_api_search_test.py 197B
task_test.py 358B
core_test.py 944B
utils
file_server.py 2KB
jwt.py 526B
stat.py 268B
pic_trans.py 592B
engine.py 2KB
file_client.py 817B
config.py 4KB
dao
__init__.py 25B
feature.py 5KB
user.py 3KB
mysql.py 2KB
cosine.py 3KB
milvus.py 4KB
docker-compose.yml 1KB
docs
deploy.md 4KB
pic
add_dir_status.png 19KB
feedback.png 23KB
login.png 23KB
add_dir.png 15KB
register.png 28KB
search.png 34KB
vue_adddir.png 37KB
vue_search.png 1.64MB
add_pic.png 15KB
engine_manage.png 19KB
design.md 5KB
search.png 31KB
api.md 10KB
add_pic.png 30KB
server.py 1KB
core
vit.py 1KB
utils.py 2KB
__init__.py 636B
lbp.py 7KB
color.py 2KB
model.py 2KB
vgg.py 891B
glcm.py 1KB
manage
fusion.py 2KB
engine_manage.py 3KB
db_manage.py 4KB
requirements.txt 292B
.gitignore 742B
conf
conf.ini 524B
README.md 3KB
共 69 条
- 1
资源评论
- 伊亦苡2024-04-07作者大大,已下载可以私聊吗
- qq_577195752024-05-21资源不错,内容挺好的,有一定的使用价值,值得借鉴,感谢分享。
mldxxxxll5
- 粉丝: 3569
- 资源: 884
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功