确保图片压缩都采用F.interpolate(),这样和安卓端得到的结果基本差不多。

2025-10-10 10:58:59 +08:00
parent 86a4f1ff67
commit 9eed8acb71
4 changed files with 17 additions and 13 deletions
@@ -468,7 +468,7 @@ class FAISSSearcher:
 def main():
    """主函数"""
    # 配置参数
-    MODEL_PATH = "model/embedding_20250917_145342/best_embedding_model.pth"
+    MODEL_PATH = "model/embedding_20250930_102826/best_embedding_model.pth"
    TRAIN_DIR = "dataset/train"
    OUTPUT_DIR = "faiss_vector_db/faiss_index"
    BATCH_SIZE = 16