修改配置路径

2025-10-11 17:39:57 +08:00
parent 2660d503d6
commit b62a64033a
4 changed files with 14 additions and 7 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]