AI贴图查找工具

基于VIT与Ball Tree的相似贴图匹配模型

本工具旨在通过计算机视觉技术与高效相似性检索算法，实现贴图数据库的快速精准匹配。用户可以输入一张本地贴图，或随机抽取一张测试图像，决定相关参数，在220张贴图的数据集中查找相似贴图。用户可以点开查找到的贴图，进入它们的本地位置进行进一步操作。用户需要决定两个参数，一个是返回相似图片的数量，一个是查找精度及速度，后者主要决定了随机投影时降至的维度数

本工具使用python开发，基于Huggingface Datasets与Transformers，使用在ImageNet21k上预训练过的 Vision Transformer (ViT) 对候选集与测试图像进行特征提取，然后通过Ball Tree 方法进行基于欧几里得距离的相似性检索，来得到最终的结果。