请求推荐适用于图片二分类相似度判断任务的AI算法

项目核心需求 #

• 文件夹a、b合计1000张独立的人物/物体图片，需完成：输入单张图片，判定其与a或b中图片更相似（属于二分类任务）
• 已在Huggingface创建模型，Model ID：2043767063，训练CO₂排放量：1.6736克

模型算法查询步骤 #

要确认这个模型使用的算法，直接从Huggingface仓库里查看以下内容即可：

1. 查看Model Card：仓库首页的说明卡通常会明确标注模型架构（如ViT、ResNet、MobileNet等）、训练框架信息
2. 读取config.json文件：文件中的model_type字段会直接给出算法类型，是最准确的判断依据
3. 检查训练脚本（若有上传）：脚本内的模型初始化代码能直观看到所使用的算法实现

项目优化思路补充 #

如果后续想优化相似性判断逻辑，也可以尝试图像特征提取+余弦相似度比对的方案：先用预训练模型提取所有图片的特征向量，输入新图片后提取其特征，分别与a、b文件夹的特征集合计算平均相似度，取相似度更高的组作为判定结果，这种方式有时比直接训练二分类模型更直观。

内容的提问来源于stack exchange，提问作者xcodertanks