文本提示图像搜索

作者：Tony Assi

使用文本提示在文件夹中搜索图像。它可以搜索文件夹中的子文件夹。基于 🤗 Transformers 和 🤗 Datasets 构建。

我们将使用我在 Github 上的 ImageSearch 模块。

下载

从 Github 下载代码

git clone https://github.com/TonyAssi/ImageSearch.git
cd ImageSearch

安装

安装所需的库

pip install -r requirements.txt

使用

导入模块

from ImageSearch import ImageSearch

定义文件夹路径

• image_dir 父文件夹
• hf_key HuggingFace 写入访问令牌可在此处创建：此处。

image_search = ImageSearch(image_dir='images',
               hf_key='HF_KEY')

首次调用此函数时，图像将被转换为 🤗 数据集并上传到 🤗 Hub。它将打印数据集 ID 并将其存储在 meta.text 文件中。它应该看起来像 tonyassi/images-ds。第一次调用会稍微慢一些。数据集创建后应该会非常快。

使用文本提示搜索图像

• text 文本提示
• download_path 与文本提示最相似的图像将下载到此路径（如果 download_path=''，则不会下载图像）
• num 图像数量（可选），默认为 5

image_search.search(text='red rose',
            download_path = 'found_images',
            num=5)

最相似的图像将被打印并下载。

flowers/0092.png
Score: 142.48643
Index: 0

flowers/0105.png
Score: 146.4621
Index: 1

flowers/0095.png
Score: 148.10144
Index: 2