修改配置

2025-10-09 09:59:36 +08:00
parent f3e3bd048a
commit a14aa7f8e1
4 changed files with 4 additions and 3 deletions
@@ -27,7 +27,7 @@ for class_name in os.listdir(input_root):
    input_dir = os.path.join(input_root, class_name)
    output_dir = os.path.join(output_root, class_name)
    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_paths = [os.path.join(input_dir, f) for f in img_files]