界面中文化

.gitignore

@@ -0,0 +1 @@
+__pycache__

app.py

@@ -8,7 +8,7 @@ from timeit import default_timer as timer
 from typing import Tuple, Dict
 
 # Setup class names
-with open("class_names.txt", "r") as f: # reading them in from class_names.txt
+with open("class_names_chinese.txt", "r") as f: # reading them in from class_names.txt
     class_names = [food_name.strip() for food_name in  f.readlines()]
     
 ### 2. Model and transforms preparation ###    
@@ -53,8 +53,8 @@ def predict(img) -> Tuple[Dict, float]:
     vit.eval()
     with torch.inference_mode():
         # Pass the transformed image through the model and turn the prediction logits into prediction probabilities
-        pred_probs = torch.softmax((effnetb2(img)+vit(img))/2, dim=1)
-    
+        pred_probs = torch.softmax((effnetb2(img_effnetb2)+vit(img_vit))/2, dim=1)
+        
     #with torch.inference_mode():
         # Pass the transformed image through the model and turn the prediction logits into prediction probabilities
         #pred_probs_vit = torch.softmax(vit(img), dim=1)
@@ -73,8 +73,11 @@ def predict(img) -> Tuple[Dict, float]:
 
 # Create title, description and article strings
 #title = "FoodVision Big 🍔👁"
-description = "An EfficientNetB2 feature extractor computer vision model to classify images of food into [101 different classes](https://github.com/mrdbourke/pytorch-deep-learning/blob/main/extras/food101_class_names.txt)."
-article = "Created at [09. PyTorch Model Deployment](https://www.learnpytorch.io/09_pytorch_model_deployment/)."
+description = "训练集使用的food101数据集，其中包含101种不同类型食品，\
+    [原项目](https://www.learnpytorch.io/09_pytorch_model_deployment/)在测试集上的精度大约为60%，\
+    我这边主要简单的替换了其中的分类模型，使得精度提到80%以上,同时也进了中文化处理，\
+    注意此分类器只包含了101个品种的食物，如披萨，饺子，炸薯条，炒饭，巧克力慕斯等等，[详细品类详见此处](class_names_chinese.txt)."
+article = "Created at [09. PyTorch Model Deployment]."
 
 title = "食品分类器 🍔👁"
 

class_names_chinese.txt

@@ -0,0 +1,101 @@
+苹果派
+小排骨
+巴卡拉
+生牛肉片
+鞑靼牛肉
+甜菜沙拉
+贝奈特饼
+拌饭
+面包布丁
+早餐卷饼
+布鲁斯凯塔
+凯撒沙拉
+香炸奶酪卷
+卡普雷塞沙拉
+胡萝卜蛋糕
+酸橘汁腌鱼
+奶酪盘
+乳酪蛋糕
+咖喱鸡
+鸡肉玉米饼
+鸡翅
+巧克力蛋糕
+巧克力慕斯
+西班牙油条
+蛤蜊杂烩
+俱乐部三明治
+蟹饼
+焦糖布丁
+法式夫人
+纸杯蛋糕
+魔鬼蛋
+甜甜圈
+水饺
+毛豆
+班尼迪克蛋
+田螺
+沙拉三明治
+烤里脊肉片
+鱼和薯条
+鹅肝
+炸薯条
+法式洋葱汤
+法式吐司
+炸鱿鱼
+炒饭
+冰冻酸奶
+大蒜面包
+汤团
+希腊式沙拉
+烤奶酪三明治
+烤三文鱼
+鳄梨
+饺子
+汉堡包
+酸辣汤
+热狗
+墨西哥煎蛋
+鹰嘴豆泥
+冰淇淋
+烤宽面条
+龙虾浓汤
+龙虾卷三明治
+通心粉和奶酪
+马卡龙
+味噌汤
+青口贝
+玉米片
+煎蛋卷
+洋葱圈
+生蚝
+泰语垫
+西班牙海鲜饭
+薄煎饼
+意式奶冻
+北京烤鸭
+河粉
+比萨
+猪排
+肉汁奶酪薯条
+牛排
+拉猪肉三明治
+拉面
+馄饨
+红色天鹅绒蛋糕
+烩饭
+萨莫萨三角饺
+生鱼片
+扇贝
+海藻沙拉
+虾和粉打窝沙食
+肉酱意粉
+意粉培根蛋面
+春卷
+牛扒
+草莓脆饼
+寿司
+炸玉米饼
+章鱼烧
+提拉米苏
+金枪鱼鞑靼
+威化饼