增加了数据增强离线的程序（可扩充数据集），将图片压缩从32*32，调整为224*224.

2025-09-15 17:26:14 +08:00
parent d73331010c
commit 264d01e9b9
5 changed files with 80 additions and 14 deletions
@@ -30,7 +30,7 @@ class FoodClassifierApp:
        self.root.geometry("1400x800")
        
        # 食物类别（根据您的数据集）
-        self.food_classes = ["回锅肉", "西红柿鸡蛋","麻辣小面"]
+        self.food_classes = ['回锅肉', '炒细面', '西红柿鸡蛋', '麻辣小面']
        
        # 设备设置
        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")