查询图片（根据自己的图片库）

准备图片数据

可使用图片数据：https://public.ukp.informatik.tu-darmstadt.de/reimers/sentence-transformers/datasets/unsplash-25k-photos.zip

也可使用自己的图片

图片数据处理

## 获取文件列表
def get_flist(path, ext, ext2=None):results = []for root, dirs, files in os.walk(path):for f in files:if f.endswith(ext):results.append(os.path.join(root, f))if ext2 is not None and f.endswith(ext2):results.append(os.path.join(root, f))return results## 获取图片库信息
def get_img(emb_file, img_path = None):  ## 判断是否存在处理好的文件（图片库信息）；有着直接读取，无着编码并存储         if os.path.exists(emb_file):with open(emb_file, 'rb') as fIn:img_names, img_emb = pickle.load(fIn)  print("Images:", len(img_names))return img_names, img_embelse:# 图片处理模型img_model = SentenceTransformer('clip-ViT-B-32')img_names = get_flist(img_path, ".jpg", ".png")print("Load Images:", len(img_names))# 只存储相对路径（便以图片地址改变）img_names2 = []for img in img_names:img_names2.append(img.replace(img_path, ""))img_emb = img_model.encode([Image.open(filepath) for filepath in img_names], batch_size=128, convert_to_tensor=True, show_progress_bar=True)with open(emb_file, 'wb') as f:pickle.dump((img_names2, img_emb), f)return img_names2, img_emb

查询图片并展示

# 查询
def search(model, img_emb, query, k=3):query_emb = model.encode([query], convert_to_tensor=True, show_progress_bar=False)hits = util.semantic_search(query_emb, img_emb, top_k=k)[0]print("Query:", query)return hits# 展示图片
def image_show(hits, img_names, img_folder):for hit in hits:img_file = img_folder + img_names[hit['corpus_id']]img = cv2.imread(img_file)cv2.imshow(str(hit['score']), img)if len(hits) > 0:cv2.waitKey(0)# 加载模型
def get_model():model = SentenceTransformer('clip-ViT-B-32-multilingual-v1')return model## 图片库地址
img_path = "E:\\image"
## 图片处理后保存的文件（便于下次直接加载）
emb_file = "E:\\image\\embeddings.pkl"
model = get_model()
img_names, img_emb = get_img(emb_file, img_path=img_path)hits = search(model, img_emb, "两只猫在睡觉")
print(hits)
image_show(hits, img_names, img_path)

结果：

Images: 24996
Query: 两只猫在睡觉
[{'corpus_id': 517, 'score': 0.30299052596092224}, {'corpus_id': 16452, 'score': 0.30096572637557983}, {'corpus_id': 9728, 'score': 0.29650789499282837}]

查询图片（根据自己的图片库）相关推荐

【Android 文件管理】分区存储 ( 创建与查询图片文件 )
文章目录一.分区存储模式下使用 MediaStore 插入图片二.分区存储模式下使用 MediaStore 查询图片三.相关文档资料 Android 分区存储系列博客 : [Android 文件 ...
OpenCV FLANN在数据集中搜索查询图片的实例(附完整代码)
OpenCV FLANN在数据集中搜索查询图片的实例 OpenCV FLANN在数据集中搜索查询图片的实例 OpenCV FLANN在数据集中搜索查询图片的实例 #include <iostre ...
java查询图片前台展示_Java +spring MVC 后台图片流在前台页面显示
用户上传图片后 (头像 or 个人照片等) ,需要在页面上显示,现在提供一种显示方式. 后台采用spring mvc的架构前台是html Java代码: @ResponseBody @Request ...
【opencv450-samples】flann_search_dataset.cpp在数据集中搜索查询图片说明 FLANN 使用的简单程序
关键点匹配效果一.数据集二.目标搜索图像三.源码: #define _CRT_SECURE_NO_WARNINGS // flann_search_dataset.cpp // Naive pr ...
Android4.4 ContentResolver查询图片无效及图库删除添加图片后，ContentResolver不更新的问题解决...
问题背景: 參考链接做了一个图片浏览,用ContentResolver扫描图库照片.且严格依照时间拍摄顺序排好序显示在listview里.例如以下图所看到的: 遇到的问题是在4.2的手机上能正常显示 ...
你生日那天的宇宙有多美？Python制作小工具一键查询图片！
这两天微博有个热搜--你生日那天的宇宙,看上去很有趣,点进去链接发现是 NASA(美国航空航天局) 提供的一个网站(庆祝哈勃望远镜30周年),可以查询一下你生日那天哈勃望远镜拍到的太空图片.但是大概是 ...
用计算机浏览电子图片,电脑制作电子图片库手机上放有什么软件
可以通过数码大师制作成音乐相册然后在手机播放的.当然,可以生成MP4等手机支持播放的格式的. 使用数码大师制作电子音乐相册: 1.导入制作电子相册的照片.首先,在顶部切换到"视频相册&quo ...
虚拟化服务器图片,虚拟化 VMware服务器图片库.ppt
虚拟化 VMware服务器图片库 VMware Infrastructure 3 Storage vMotion VMware Infrastructure 3 Update Manager VMw ...
Linux下查询图片的大小
方法一:使用file命令(但是 jpg 文件就不太好用.) [root@server1 download]# ls vim.jpg [root@server1 download]# file vim. ...

查询图片（根据自己的图片库）

准备图片数据

图片数据处理

查询图片并展示

查询图片（根据自己的图片库）相关推荐

最新文章

热门文章