基于OpenCV的图象RGB直方图检索系统
文件结构:
DB/ 数据库文件夹(存放图像的直方图特征)
mirflickr/ 图片文件夹(存放jpg图片,本例为mirflickr25k数据集)
config.py 配置文件(定义图片文件夹路径、数据库文件夹路径、图像采样大小、图像分块大小、直方图桶大小、图像颜色通道数、相似度阈值)
db.py 数据库程序
main.py 图像检索主程序
test.jpg 需要在数据库中检索的图片
util.py 通用的操作库
运行环境:
Python 2
Numpy
Matplotlib
cv2
环境配置:
下载并安装anaconda2
在命令行中运行
conda install -c menpo opencv3
以安装opencv for python
运行流程:
下载数据集(http://press.liacs.nl/mirflickr/mirflickr25k.v2/mirflickr25k.zip),将其中的jpg文件放入图片文件夹
*配置config.py(如果使用自己的数据集),主要修改图片文件夹路径和相似度阈值
*修改db.py中的备注选项以可视化数据库创建过程和直方图计算结果
删除数据库文件夹中的所有内容
运行
python db.py
以创建图像数据文件库(创建好后,无需第二次运行)
将想要检索的图片放在目录下以test.jpg命名
运行
python main.py
所有超过相似度阈值的图片将以窗口形式显示出来
1