如何利用小模型,快速筛选照片
介绍
检测重复图片,识别相似图片,并进行图片分类。
最近在整理照片,由于照片的量太多了,并且有很多类似的照片,我希望能够将这一堆照片中的重复图片清理掉,同时将类似的,或者比较相近的照片放到一起。
比如,在同一个场景,或者穿着同一套衣服的照片,希望能够集中整理。
然而,我的台式机上并没有一张很好的显卡,显然跑不了VL模型,我想大多数人的电脑上,应该都没有能够跑动VL模型的显卡。
好在,我只要简单的做一些图片的相似度检测。
在本项目中,我用到了一些CNN模型,这些模型比较小,可以在常规的CPU上进行推理,效率也很快,我处理700张照片,大概5-10分钟,你可以更快。
项目地址: github传送门
使用
本项目没有WEBUI界面,也没有封装的执行文件,只有一些脚本,配合python环境运行。如果你对执行文件编译比较在行,也欢迎将我的代码编译成可执行文件。
本项目是一个业余项目,如果存在bug,或者有一些好的点子,可以给我提issue,有时间的话,会更新。也欢迎共创。
快速开始
本项目没有UI,只能配置好环境后,执行python脚本。
- 安装python环境
自行下载python,我用的是python-3.12。或者使用conda