Просмотр исходного кода

模型识别结果直接从索引id读取中文

张洋 1 неделя назад
Родитель
Сommit
022ace793d
2 измененных файлов с 344 добавлено и 264 удалено
  1. 342 253
      zhipuzi_pos_windows/ai/YoloClassName.h
  2. 2 11
      zhipuzi_pos_windows/ai/YoloFeatureManager.cpp

+ 342 - 253
zhipuzi_pos_windows/ai/YoloClassName.h

@@ -3,262 +3,351 @@
 #include <string>
 #include <string>
 #include <unordered_map>
 #include <unordered_map>
 
 
-// 果蔬名称常量数组(索引对应英文名称)
-// 注释格式:[索引] 英文名称 // 中文名称
+//记录当前使用的模型的版本号,如果模型升级了,所有向量数据库里面的版本号不匹配了,就需要重新生成向量数据库了
+const std::string FRUIT_VEGETABLE_AI_MODEL_VERSION = "1.0.0";
+
+// 果蔬名称常量数组,yolo识别的索引在这里根据索引获取商品名字
+// 注释格式:[索引] 中文名称
 const std::string FRUIT_VEGETABLE_NAMES[] = {
 const std::string FRUIT_VEGETABLE_NAMES[] = {
-    "apple_akosu",          // 0: 阿克苏苹果
-    "apple_gala",           // 1: 嘎啦苹果
-    "apple_green",          // 2: 青苹果
-    "apple_hua_niu",        // 3: 花牛苹果
-    "apple_red_fuji",       // 4: 红富士苹果
-    "apricot",              // 5: 杏
-    "asparagus",            // 6: 芦笋
-    "avocado",              // 7: 牛油果
-    "banana",               // 8: 香蕉
-    "bamboo_shoot",         // 9: 竹笋
-    "baby_cabbage",         // 10: 娃娃菜
-    "bean_green",           // 11: 四季豆
-    "bitter_gourd",         // 12: 苦瓜
-    "blueberry",            // 13: 蓝莓
-    "bokchoy",              // 14: 上海青
-    "broccoli",             // 15: 西兰花
-    "cabbage_chinese",      // 16: 大白菜
-    "cabbage_purple",       // 17: 紫甘蓝
-    "cabbage_round",        // 18: 圆白菜
-    "cabbage_small",        // 19: 小白菜
-    "cantaloupe",           // 20: 哈密瓜
-    "cantaloupe_nettle",    // 21: 网纹瓜
-    "cantaloupe_xizhou",    // 22: 西州蜜
-    "carrot",               // 23: 胡萝卜
-    "carrot_yellow",        // 24: 黄胡萝卜
-    "cauliflower",          // 25: 白花菜
-    "caixin",               // 26: 菜心
-    "celery",               // 27: 芹菜
-    "cherry",               // 28: 樱桃
-    "cherry_big",           // 29: 车厘子
-    "chili_small",          // 30: 小米辣
-    "chinese_yam",          // 31: 山药
-    "chives",               // 32: 韭菜
-    "coriander",            // 33: 香菜
-    "corn",                 // 34: 玉米
-    "cowpea",               // 35: 豇豆
-    "cucumber",             // 36: 黄瓜
-    "cucumber_fruit",       // 37: 水果黄瓜
-    "cucumber_white",       // 38: 白玉黄瓜
-    "durian",               // 39: 榴莲
-    "dragonfruit_red",      // 40: 红心火龙果
-    "dragonfruit_white",    // 41: 白心火龙果
-    "eggplant_long",        // 42: 长茄子
-    "eggplant_round",       // 43: 圆茄子
-    "garlic",               // 44: 大蒜
-    "garlic_scape",         // 45: 蒜苔
-    "ginger",               // 46: 生姜
-    "grape_black",          // 47: 夏黑
-    "grape_green",          // 48: 青提
-    "grape_kyoho",          // 49: 巨峰葡萄
-    "grape_red",            // 50: 红提
-    "grape_seedless",       // 51: 无籽葡萄
-    "grape_shine",          // 52: 阳光玫瑰
-    "grapefruit",           // 53: 西柚
-    "jujube_winter",        // 54: 冬枣
-    "kiwi_green",           // 55: 绿心猕猴桃
-    "kiwi_red",             // 56: 红心猕猴桃
-    "lemon",                // 57: 柠檬
-    "lime",                // 58: 青柠
-    "lettuce",              // 59: 生菜
-    "lettuce_romaine",      // 60: 罗马生菜
-    "litchi",               // 61: 荔枝
-    "loofah",               // 62: 丝瓜
-    "longan",               // 63: 龙眼
-    "lotus_root",           // 64: 莲藕
-    "mango_golden",         // 65: 金煌芒
-    "mango_guifei",         // 66: 贵妃芒
-    "mango_small",          // 67: 小台农芒果
-    "mangosteen",           // 68: 山竹
-    "muskmelon_common",     // 69: 香瓜
-    "muskmelon_green",      // 70: 绿宝甜瓜
-    "muskmelon_horn",       // 71: 羊角蜜
-    "mushroom_enoki",       // 72: 金针菇
-    "mushroom_king",        // 73: 杏鲍菇
-    "mushroom_oyster",      // 74: 平菇
-    "mushroom_shiitake",    // 75: 香菇
-    "okra",                 // 76: 秋葵
-    "oil_vegetable",        // 77: 油麦菜
-    "onion_purple",         // 78: 紫洋葱
-    "onion_white",          // 79: 白洋葱
-    "orange_mandarin",      // 80: 蜜橘
-    "orange_navel",         // 81: 脐橙
-    "orange_papa",          // 82: 耙耙柑
-    "orange_tangerine",     // 83: 砂糖橘
-    "orange_ugly",          // 84: 丑橘
-    "orange_wogan",         // 85: 沃柑
-    "papaya",               // 86: 木瓜
-    "peach_flat",           // 87: 蟠桃
-    "peach_honey",          // 88: 水蜜桃
-    "peach_nectarine",      // 89: 油桃
-    "peach_yellow",         // 90: 黄桃
-    "pear_kuerle",          // 91: 库尔勒香梨
-    "pear_qiuyue",          // 92: 秋月梨
-    "pear_snow",            // 93: 雪梨
-    "pear_ya",              // 94: 鸭梨
-    "pepper_bell_green",    // 95: 绿彩椒
-    "pepper_bell_red",      // 96: 红彩椒
-    "pepper_bell_yellow",   // 97: 黄彩椒
-    "pea",                  // 98: 豌豆
-    "pineapple",            // 99: 菠萝
-    "pineapple_yellow",     // 100: 凤梨
-    "plum",                 // 101: 李子
-    "plum_black",           // 102: 黑布林
-    "pomegranate",          // 103: 石榴
-    "potato",               // 104: 土豆
-    "pumpkin",              // 105: 南瓜
-    "pumpkin_baby",         // 106: 贝贝南瓜
-    "purple_potato",        // 107: 紫薯
-    "radish_cherry",        // 108: 樱桃萝卜
-    "radish_red",           // 109: 红萝卜
-    "radish_white",         // 110: 白萝卜
-    "spinach",              // 111: 菠菜
-    "strawberry",           // 112: 草莓
-    "sweet_potato",         // 113: 红薯
-    "tomato_big",           // 114: 大番茄
-    "tomato_cherry",        // 115: 圣女果
-    "tomato_qianxi",        // 116: 千禧果
-    "wax_gourd",            // 117: 冬瓜
-    "watermelon",           // 118: 西瓜
-    "watermelon_8424",      // 119: 8424西瓜
-    "watermelon_black_beauty", // 120: 黑美人西瓜
-    "watermelon_qilin",     // 121: 麒麟西瓜
-    "wood_ear",             // 122: 木耳
-    "zucchini"              // 123: 西葫芦
+	"红富士苹果",        // 0
+	"嘎啦苹果",          // 1
+	"青苹果",            // 2
+	"阿克苏苹果",        // 3
+	"花牛苹果",          // 4
+	"秦安苹果",          // 5
+	"蛇果",              // 6
+	"国光苹果",          // 7
+	"红星苹果",          // 8
+	"黄元帅苹果",        // 9
+	"乔纳金苹果",        // 10
+	"鸡心果",            // 11
+	"鸭梨",              // 12
+	"库尔勒香梨",        // 13
+	"雪梨",              // 14
+	"蜜梨",              // 15
+	"秋月梨",            // 16
+	"丰水梨",            // 17
+	"皇冠梨",            // 18
+	"酥梨",              // 19
+	"莱阳梨",            // 20
+	"黄金梨",            // 21
+	"啤梨",              // 22
+	"南果梨",            // 23
+	"京白梨",            // 24
+	"早酥梨",            // 25
+	"海棠果",            // 26
+	"沙果",              // 27
+	"山楂",              // 28
+	"山里红",            // 29
+	"水蜜桃",            // 30
+	"黄桃",              // 31
+	"蟠桃",              // 32
+	"油桃",              // 33
+	"毛桃",              // 34
+	"血桃",              // 35
+	"脆桃",              // 36
+	"鹰嘴桃",            // 37
+	"冬桃",              // 38
+	"李子",              // 39
+	"黑布林",            // 40
+	"红布林",            // 41
+	"青李子",            // 42
+	"蜂糖李",            // 43
+	"杏",                // 44
+	"油杏",              // 45
+	"梅子",              // 46
+	"杨梅",              // 47
+	"西梅",              // 48
+	"樱桃",              // 49
+	"车厘子",            // 50
+	"冬枣",              // 51
+	"青枣",              // 52
+	"脆枣",              // 53
+	"沙枣",              // 54
+	"酸枣",              // 55
+	"大枣",              // 56
+	"荔枝",              // 57
+	"龙眼",              // 58
+	"黄皮",              // 59
+	"橄榄",              // 60
+	"枇杷",              // 61
+	"青梅",              // 62
+	"脐橙",              // 63
+	"冰糖橙",            // 64
+	"血橙",              // 65
+	"蜜橘",              // 66
+	"砂糖橘",            // 67
+	"沃柑",              // 68
+	"耙耙柑",            // 69
+	"丑橘",              // 70
+	"皇帝柑",            // 71
+	"金桔",              // 72
+	"脆皮金桔",          // 73
+	"西柚",              // 74
+	"文旦",              // 75
+	"胡柚",              // 76
+	"蜜柚",              // 77
+	"椪柑",              // 78
+	"茂谷柑",            // 79
+	"黄柠檬",            // 80
+	"青柠",              // 81
+	"香水柠檬",          // 82
+	"无籽柠檬",          // 83
+	"红提",              // 84
+	"青提",              // 85
+	"黑提",              // 86
+	"巨峰葡萄",          // 87
+	"阳光玫瑰",          // 88
+	"夏黑",              // 89
+	"无籽葡萄",          // 90
+	"玫瑰香",            // 91
+	"马奶葡萄",          // 92
+	"水晶葡萄",          // 93
+	"美人指",            // 94
+	"蓝宝石葡萄",        // 95
+	"草莓",              // 96
+	"蓝莓",              // 97
+	"桑葚",              // 98
+	"黑加仑",            // 99
+	"姑娘果",            // 100
+	"树莓",              // 101
+	"猕猴桃",            // 102
+	"软枣猕猴桃",        // 103
+	"蔓越莓鲜果",        // 104
+	"西瓜",              // 105
+	"黑美人西瓜",        // 106
+	"8424西瓜",          // 107
+	"麒麟西瓜",          // 108
+	"甜王西瓜",          // 109
+	"京欣西瓜",          // 110
+	"特小凤西瓜",        // 111
+	"哈密瓜",            // 112
+	"西州蜜",            // 113
+	"网纹瓜",            // 114
+	"伊丽莎白瓜",       // 115
+	"甜瓜",              // 116
+	"羊角蜜",            // 117
+	"绿宝甜瓜",          // 118
+	"东方蜜",            // 119
+	"香瓜",              // 120
+	"白兰瓜",            // 121
+	"羊角脆",            // 122
+	"玉菇甜瓜",          // 123
+	"阎良甜瓜",          // 124
+	"黄河蜜瓜",          // 125
+	"香蕉",              // 126
+	"皇帝蕉",            // 127
+	"粉蕉",              // 128
+	"芭蕉",              // 129
+	"菠萝",              // 130
+	"凤梨",              // 131
+	"金煌芒",            // 132
+	"小台农芒果",        // 133
+	"贵妃芒",            // 134
+	"凯特芒",            // 135
+	"象牙芒",            // 136
+	"青芒",              // 137
+	"澳芒",              // 138
+	"桂七芒",            // 139
+	"石榴",              // 140
+	"软籽石榴",          // 141
+	"无花果",            // 142
+	"山竹",              // 143
+	"莲雾",              // 144
+	"红毛丹",            // 145
+	"释迦",              // 146
+	"杨桃",              // 147
+	"百香果",            // 148
+	"番石榴",            // 149
+	"人参果",            // 150
+	"菠萝蜜",            // 151
+	"火龙果",            // 152
+	"牛油果",            // 153
+	"木瓜",              // 154
+	"椰子",              // 155
+	"椰青",              // 156
+	"椰皇",              // 157
+	"柿子",              // 158
+	"脆柿",              // 159
+	"软柿",              // 160
+	"余甘果",            // 161
+	"榴莲",              // 162
+	"火晶柿子",          // 163
+	"甘蔗",              // 164
+	"雪莲果",            // 165
+	"沙棘",              // 166
+	"荸荠",              // 167
+	"大白菜",            // 168
+	"小白菜",            // 169
+	"上海青",            // 170
+	"菜心",              // 171
+	"娃娃菜",            // 172
+	"乌塌菜",            // 173
+	"鸡毛菜",            // 174
+	"奶白菜",            // 175
+	"圆白菜",            // 176
+	"紫甘蓝",            // 177
+	"西兰花",            // 178
+	"白花菜",            // 179
+	"松花菜",            // 180
+	"芥蓝",              // 181
+	"苤蓝",              // 182
+	"羽衣甘蓝",          // 183
+	"孢子甘蓝",          // 184
+	"白萝卜",            // 185
+	"红萝卜",            // 186
+	"青萝卜",            // 187
+	"樱桃萝卜",          // 188
+	"心里美萝卜",        // 189
+	"胡萝卜",            // 190
+	"黄胡萝卜",          // 191
+	"水果胡萝卜",        // 192
+	"芜菁",              // 193
+	"牛蒡",              // 194
+	"水萝卜",            // 195
+	"甜菜根",            // 196
+	"山药",              // 197
+	"铁棍山药",          // 198
+	"土豆",              // 199
+	"紫薯",              // 200
+	"红薯",              // 201
+	"蜜薯",              // 202
+	"芋头",              // 203
+	"魔芋",              // 204
+	"菱角",              // 205
+	"莲藕",              // 206
+	"慈姑",              // 207
+	"生姜",              // 208
+	"凉薯",              // 209
+	"沙姜",              // 210
+	"菠菜",              // 211
+	"油麦菜",            // 212
+	"生菜",              // 213
+	"罗马生菜",         // 214
+	"奶油生菜",         // 215
+	"莴笋",              // 216
+	"空心菜",            // 217
+	"木耳菜",            // 218
+	"茼蒿",              // 219
+	"茴香",              // 220
+	"芹菜",              // 221
+	"西芹",              // 222
+	"香菜",              // 223
+	"苋菜",              // 224
+	"苦苣",              // 225
+	"香椿",              // 226
+	"薄荷",              // 227
+	"紫苏",              // 228
+	"芥菜",              // 229
+	"豌豆尖",            // 230
+	"红菜苔",            // 231
+	"地瓜叶",            // 232
+	"蒲公英",            // 233
+	"荠菜",              // 234
+	"枸杞苗",            // 235
+	"莼菜",              // 236
+	"蒿子秆",            // 237
+	"西洋菜",            // 238
+	"折耳根",            // 239
+	"紫背天葵",          // 240
+	"马齿苋",            // 241
+	"大葱",              // 242
+	"小葱",              // 243
+	"韭菜",              // 244
+	"韭黄",              // 245
+	"韭苔",              // 246
+	"蒜苔",              // 247
+	"蒜苗",              // 248
+	"藠头",              // 249
+	"大蒜",              // 250
+	"蒜黄",              // 251
+	"韭菜花",            // 252
+	"白洋葱",            // 253
+	"紫洋葱",            // 254
+	"黄洋葱",            // 255
+	"红彩椒",            // 256
+	"黄彩椒",            // 257
+	"绿彩椒",            // 258
+	"小米辣",            // 259
+	"朝天椒",            // 260
+	"线椒",              // 261
+	"杭椒",              // 262
+	"甜椒",              // 263
+	"泡椒",              // 264
+	"螺丝椒",            // 265
+	"二荆条",            // 266
+	"野山椒",            // 267
+	"大番茄",            // 268
+	"圣女果",            // 269
+	"千禧果",            // 270
+	"长茄子",            // 271
+	"圆茄子",            // 272
+	"青茄子",            // 273
+	"紫茄子",            // 274
+	"秋葵",              // 275
+	"彩茄",              // 276
+	"黄瓜",              // 277
+	"白玉黄瓜",          // 278
+	"水果黄瓜",          // 279
+	"旱黄瓜",            // 280
+	"冬瓜",              // 281
+	"南瓜",              // 282
+	"贝贝南瓜",          // 283
+	"板栗南瓜",          // 284
+	"丝瓜",              // 285
+	"苦瓜",              // 286
+	"佛手瓜",            // 287
+	"西葫芦",            // 288
+	"瓠瓜",              // 289
+	"节瓜",              // 290
+	"赖瓜",              // 291
+	"绿豆芽",            // 292
+	"黄豆芽",            // 293
+	"四季豆",            // 294
+	"豇豆",              // 295
+	"豌豆",              // 296
+	"荷兰豆",            // 297
+	"毛豆",              // 298
+	"扁豆",              // 299
+	"刀豆",              // 300
+	"蚕豆",              // 301
+	"豌豆苗",            // 302
+	"花生芽",            // 303
+	"芸豆",              // 304
+	"芦笋",              // 305
+	"春笋",              // 306
+	"冬笋",              // 307
+	"竹笋",              // 308
+	"茭白",              // 309
+	"金针菜",            // 310
+	"鲜百合",            // 311
+	"玉米",              // 312
+	"玉米笋",            // 313
+	"黑糯玉米",          // 314
+	"平菇",              // 315
+	"香菇",              // 316
+	"金针菇",            // 317
+	"杏鲍菇",            // 318
+	"蟹味菇",            // 319
+	"白玉菇",            // 320
+	"海鲜菇",            // 321
+	"木耳",              // 322
+	"银耳",              // 323
+	"猴头菇",            // 324
+	"茶树菇",            // 325
+	"鸡腿菇",            // 326
+	"草菇",              // 327
+	"口蘑",              // 328
+	"竹荪",              // 329
+	"滑子菇",            // 330
+	"鹿茸菇",            // 331
+	"鲜海带",            // 332
+	"裙带菜",            // 333
+	"海白菜"             // 334
 };
 };
 
 
-// 果蔬英文名称 -> 中文名称 映射表(哈希表,查询效率O(1))
-const std::unordered_map<std::string, std::string> FRUIT_VEGETABLE_CN_MAP = {
-    {"apple_akosu", "阿克苏苹果"},
-    {"apple_gala", "嘎啦苹果"},
-    {"apple_green", "青苹果"},
-    {"apple_hua_niu", "花牛苹果"},
-    {"apple_red_fuji", "红富士苹果"},
-    {"apricot", "杏"},
-    {"asparagus", "芦笋"},
-    {"avocado", "牛油果"},
-    {"banana", "香蕉"},
-    {"bamboo_shoot", "竹笋"},
-    {"baby_cabbage", "娃娃菜"},
-    {"bean_green", "四季豆"},
-    {"bitter_gourd", "苦瓜"},
-    {"blueberry", "蓝莓"},
-    {"bokchoy", "上海青"},
-    {"broccoli", "西兰花"},
-    {"cabbage_chinese", "大白菜"},
-    {"cabbage_purple", "紫甘蓝"},
-    {"cabbage_round", "圆白菜"},
-    {"cabbage_small", "小白菜"},
-    {"cantaloupe", "哈密瓜"},
-    {"cantaloupe_nettle", "网纹瓜"},
-    {"cantaloupe_xizhou", "西州蜜"},
-    {"carrot", "胡萝卜"},
-    {"carrot_yellow", "黄胡萝卜"},
-    {"cauliflower", "白花菜"},
-    {"caixin", "菜心"},
-    {"celery", "芹菜"},
-    {"cherry", "樱桃"},
-    {"cherry_big", "车厘子"},
-    {"chili_small", "小米辣"},
-    {"chinese_yam", "山药"},
-    {"chives", "韭菜"},
-    {"coriander", "香菜"},
-    {"corn", "玉米"},
-    {"cowpea", "豇豆"},
-    {"cucumber", "黄瓜"},
-    {"cucumber_fruit", "水果黄瓜"},
-    {"cucumber_white", "白玉黄瓜"},
-    {"durian", "榴莲"},
-    {"dragonfruit_red", "红心火龙果"},
-    {"dragonfruit_white", "白心火龙果"},
-    {"eggplant_long", "长茄子"},
-    {"eggplant_round", "圆茄子"},
-    {"garlic", "大蒜"},
-    {"garlic_scape", "蒜苔"},
-    {"ginger", "生姜"},
-    {"grape_black", "夏黑"},
-    {"grape_green", "青提"},
-    {"grape_kyoho", "巨峰葡萄"},
-    {"grape_red", "红提"},
-    {"grape_seedless", "无籽葡萄"},
-    {"grape_shine", "阳光玫瑰"},
-    {"grapefruit", "西柚"},
-    {"jujube_winter", "冬枣"},
-    {"kiwi_green", "绿心猕猴桃"},
-    {"kiwi_red", "红心猕猴桃"},
-    {"lemon", "柠檬"},
-    {"lime", "青柠"},
-    {"lettuce", "生菜"},
-    {"lettuce_romaine", "罗马生菜"},
-    {"litchi", "荔枝"},
-    {"loofah", "丝瓜"},
-    {"longan", "龙眼"},
-    {"lotus_root", "莲藕"},
-    {"mango_golden", "金煌芒"},
-    {"mango_guifei", "贵妃芒"},
-    {"mango_small", "小台农芒果"},
-    {"mangosteen", "山竹"},
-    {"muskmelon_common", "香瓜"},
-    {"muskmelon_green", "绿宝甜瓜"},
-    {"muskmelon_horn", "羊角蜜"},
-    {"mushroom_enoki", "金针菇"},
-    {"mushroom_king", "杏鲍菇"},
-    {"mushroom_oyster", "平菇"},
-    {"mushroom_shiitake", "香菇"},
-    {"okra", "秋葵"},
-    {"oil_vegetable", "油麦菜"},
-    {"onion_purple", "紫洋葱"},
-    {"onion_white", "白洋葱"},
-    {"orange_mandarin", "蜜橘"},
-    {"orange_navel", "脐橙"},
-    {"orange_papa", "耙耙柑"},
-    {"orange_tangerine", "砂糖橘"},
-    {"orange_ugly", "丑橘"},
-    {"orange_wogan", "沃柑"},
-    {"papaya", "木瓜"},
-    {"peach_flat", "蟠桃"},
-    {"peach_honey", "水蜜桃"},
-    {"peach_nectarine", "油桃"},
-    {"peach_yellow", "黄桃"},
-    {"pear_kuerle", "库尔勒香梨"},
-    {"pear_qiuyue", "秋月梨"},
-    {"pear_snow", "雪梨"},
-    {"pear_ya", "鸭梨"},
-    {"pepper_bell_green", "绿彩椒"},
-    {"pepper_bell_red", "红彩椒"},
-    {"pepper_bell_yellow", "黄彩椒"},
-    {"pea", "豌豆"},
-    {"pineapple", "菠萝"},
-    {"pineapple_yellow", "凤梨"},
-    {"plum", "李子"},
-    {"plum_black", "黑布林"},
-    {"pomegranate", "石榴"},
-    {"potato", "土豆"},
-    {"pumpkin", "南瓜"},
-    {"pumpkin_baby", "贝贝南瓜"},
-    {"purple_potato", "紫薯"},
-    {"radish_cherry", "樱桃萝卜"},
-    {"radish_red", "红萝卜"},
-    {"radish_white", "白萝卜"},
-    {"spinach", "菠菜"},
-    {"strawberry", "草莓"},
-    {"sweet_potato", "红薯"},
-    {"tomato_big", "大番茄"},
-    {"tomato_cherry", "圣女果"},
-    {"tomato_qianxi", "千禧果"},
-    {"wax_gourd", "冬瓜"},
-    {"watermelon", "西瓜"},
-    {"watermelon_8424", "8424西瓜"},
-    {"watermelon_black_beauty", "黑美人西瓜"},
-    {"watermelon_qilin", "麒麟西瓜"},
-    {"wood_ear", "木耳"},
-    {"zucchini", "西葫芦"}
-};
+// 可以加一个长度常量,方便代码使用
+const int FRUIT_VEGETABLE_COUNT = sizeof(FRUIT_VEGETABLE_NAMES) / sizeof(std::string);
 
 
 // 蔬菜水果中文别名映射表 (key: 标准名称, value: 别名|别名|...)
 // 蔬菜水果中文别名映射表 (key: 标准名称, value: 别名|别名|...)
 // 因为同一个水果或者蔬菜,有不同的叫法,我们模型训练的时候是按照标准名称来标注的,AI识别出来的是标准名称,但是收银系统里面录入的可能是别名不一定是标准名称,会导致搜索标准名称搜索不到这个商品
 // 因为同一个水果或者蔬菜,有不同的叫法,我们模型训练的时候是按照标准名称来标注的,AI识别出来的是标准名称,但是收银系统里面录入的可能是别名不一定是标准名称,会导致搜索标准名称搜索不到这个商品

+ 2 - 11
zhipuzi_pos_windows/ai/YoloFeatureManager.cpp

@@ -98,18 +98,9 @@ std::string YoloFeatureManager::getClassName(std::size_t classId) const
 {
 {
 	if (classId >= 0 && classId < FRUIT_VEGETABLE_COUNT)
 	if (classId >= 0 && classId < FRUIT_VEGETABLE_COUNT)
 	{
 	{
-		std::string englishName = FRUIT_VEGETABLE_NAMES[classId];
+		std::string cnName = FRUIT_VEGETABLE_NAMES[classId];
 
 
-		// 这里可以添加一个映射表,将英文名称映射到中文名称
-		auto it = FRUIT_VEGETABLE_CN_MAP.find(englishName);
-		if (it != FRUIT_VEGETABLE_CN_MAP.end())
-		{
-			return it->second; // 找到则返回中文名称
-		}
-		else
-		{
-			return "Unknown"; // 未找到则返回默认值
-		}
+		return cnName;
 	}
 	}
 
 
 	return "Unknown";
 	return "Unknown";