Browse Source

调低ai识别置信度,增加ai bus标签

lang 3 years ago
parent
commit
36be097621
1 changed files with 2 additions and 2 deletions
  1. 2 2
      Controller/AiController.py

+ 2 - 2
Controller/AiController.py

@@ -127,7 +127,7 @@ class AiView(View):
             # photo = open(r'E:\test---------------\test\snipaste20220121_215952.jpg', 'rb')
             # 识别合成图片
             maxLabels = 50  # 最大标签
-            minConfidence = 95  # 置信度
+            minConfidence = 70  # 置信度
 
             ai_start_time = int(time.time())
             client = boto3.client(
@@ -302,7 +302,7 @@ class AiView(View):
         all_labels_type = {
             '1': ['Person', 'Human'],  # 人
             '2': ['Dog', 'Pet', 'Canine', 'Animal', 'Puppy', 'Cat'],  # 动物
-            '3': ['Car', 'Vehicle', 'Transportation', 'Automobile'],  # 车
+            '3': ['Car', 'Vehicle', 'Transportation', 'Automobile', 'Bus'],  # 车
             '4': ['Carton', 'Cardboard', 'Package Delivery']  # 包裹
         }
         # 找出识别的所有标签