只是更改了一些配置

This commit is contained in:
2025-10-10 15:59:01 +08:00
parent 9eed8acb71
commit 2660d503d6
4 changed files with 6 additions and 6 deletions
+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)
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]