yixiong/FoodClassifier
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改配置路径

Browse Source
This commit is contained in:
zhangpu
2025-10-11 17:39:57 +08:00
parent 2660d503d6
commit b62a64033a
4 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
classifier/embedding_food_classifier_app.py
+6 -2
View File
@@ -61,10 +61,14 @@ class EmbeddingFoodClassifierApp:
"""加载训练好的embedding模型和FAISS索引""" """加载训练好的embedding模型和FAISS索引"""
try: try:
# 模型路径 # 模型路径
model_path = "../model/embedding_20251010_111257/best_embedding_model.pth" # 获取当前脚本所在目录
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
# model_path = "../model/embedding_20251011_133653/best_embedding_model.pth"
model_path = os.path.join(BASE_DIR, "../model/embedding_20251011_133653/best_embedding_model.pth")
# FAISS索引目录 # FAISS索引目录
index_dir = "../faiss_vector_db/faiss_index" # index_dir = "../faiss_vector_db/faiss_index"
index_dir = os.path.join(BASE_DIR, "../faiss_vector_db/faiss_index")
if os.path.exists(model_path) and os.path.exists(index_dir): if os.path.exists(model_path) and os.path.exists(index_dir):
# 1. 加载embedding模型 # 1. 加载embedding模型
classifier/embedding_food_classifier_search_app.py
+6 -3
View File
@@ -62,10 +62,13 @@ class EmbeddingFoodClassifierApp:
"""加载训练好的embedding模型和FAISS索引""" """加载训练好的embedding模型和FAISS索引"""
try: try:
# 模型路径 # 模型路径
model_path = "../model/embedding_20251010_111257/best_embedding_model.pth" # 获取当前脚本所在目录
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
# model_path = "../model/embedding_20251011_133653/best_embedding_model.pth"
model_path = os.path.join(BASE_DIR, "../model/embedding_20251011_133653/best_embedding_model.pth")
# FAISS索引目录 # FAISS索引目录
index_dir = "../faiss_vector_db/faiss_index" # index_dir = "../faiss_vector_db/faiss_index"
index_dir = os.path.join(BASE_DIR, "../faiss_vector_db/faiss_index")
if os.path.exists(model_path) and os.path.exists(index_dir): if os.path.exists(model_path) and os.path.exists(index_dir):
# 1. 加载embedding模型 # 1. 加载embedding模型
data_expansion/data_expansion.py
+1 -1
View File
@@ -27,7 +27,7 @@ for class_name in os.listdir(input_root):
input_dir = os.path.join(input_root, class_name) input_dir = os.path.join(input_root, class_name)
output_dir = os.path.join(output_root, class_name) output_dir = os.path.join(output_root, class_name)
os.makedirs(output_dir, exist_ok=True) os.makedirs(output_dir, exist_ok=True)
if class_name == "手撕包菜": if class_name == "红烧肉":
# 读取类别下所有图片路径 # 读取类别下所有图片路径
img_files = [f for f in os.listdir(input_dir) if f.lower().endswith(('.jpg', '.png', '.jpeg'))] img_files = [f for f in os.listdir(input_dir) if f.lower().endswith(('.jpg', '.png', '.jpeg'))]
img_paths = [os.path.join(input_dir, f) for f in img_files] img_paths = [os.path.join(input_dir, f) for f in img_files]
faiss_vector_db/build_faiss_index.py
+1 -1
View File
@@ -477,7 +477,7 @@ class FAISSSearcher:
def main(): def main():
"""主函数""" """主函数"""
# 配置参数 # 配置参数
MODEL_PATH = "../model/embedding_20251010_111257/best_embedding_model.pth" MODEL_PATH = "../model/embedding_20251011_133653/best_embedding_model.pth"
TRAIN_DIR = "../dataset/train" TRAIN_DIR = "../dataset/train"
OUTPUT_DIR = "faiss_index" OUTPUT_DIR = "faiss_index"
BATCH_SIZE = 16 BATCH_SIZE = 16