# imagine-search

**Repository Path**: ge-mingyu/imagine-search

## Basic Information

- **Project Name**: imagine-search
- **Description**: 以图搜图
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2021-09-23
- **Last Updated**: 2023-02-25

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

#search_engine

#### 介绍
图片搜索库，以图搜图功能。

#### 技术栈
python 3.8.8 + pytorch(未使用GPU加速) + matplotlib + numpy + h5py

#### 使用说明

1. 由于使用网络结构为VGG16，先获取对应的模型参数（.pth后缀文件），两种方法：
    - 运行 Model->DownLoadPreModel.py 文件，下方显示进度条，下载好之后会放置在同级state文件夹下，一个.pth后缀的文件
    - 下载百度云文件，然后放置到state文件夹之下（链接: https://pan.baidu.com/s/1I9_q_yieVQ8jgZFS-yuuSw 提取码: spx4）
   
2. 获取数据集，数据集放置地址 DB->pictures (已保留了一些测试用的数据图片，如有需要直接往里面放图片就行)
3. 运行 DataHandle.py文件，会自动检索 DB->pictures 下的所有 .jpg 结尾的图片并获取对应特征值与对应路径并保存为.h5文件在FeaturesAndPaths文件夹下面
4. 在 src->CNNMOdel 文件夹下放置一个名字叫 goal.jpg 的图片（就是要用于搜索的图片）
5. 完成上述之后运行SearchImage.py文件，就会获得与 goal.jpg 相似的图片

ps.
   - 通过修改 SearchImage.py 文件下的 query 变量可以改变用于搜索图片的名称
   - 通过修改 SearchImage.py 文件下的 maxres 变量可以改变获取到的相似图片的个数