深度图像散列
通过杂货产品的图像哈希快速查询搜索。
作为Digital Horizons夏季远程实习的一部分而完成。
依存关系
Python 3.6+
脾气暴躁的
火炬
Matplotlib
枕头
Tqdm
烦人的
运行以下命令以安装这些依赖项
pip install -r requirements.txt
管道
数据集详细信息
使用了Digital Horizons提供的杂货产品。 它有10种不同的产品,从肥皂到饼干。 数据的组织迫使图像文件夹结构如下所示。
由于给定的数据集不平衡,我们进行类重新平衡以使所有类具有相等数量的图像。
用法
python main.py
有关一些参数的
isTrain-是否训练模型
isSaveHash-是否在整个数据集中运行经过训练的模型以保存哈希
TRANSFORM_IMG中的规范化可能必须更改,具体取决于您的数据。 使
评论0
最新资源