|
|
@@ -152,8 +152,8 @@ class AiView(View):
|
|
|
|
|
|
# photo = open(r'E:\test---------------\test\snipaste20220121_215952.jpg', 'rb')
|
|
|
#识别合成图片
|
|
|
- maxLabels = 50
|
|
|
- minConfidence = 70
|
|
|
+ maxLabels = 50 #最大标签
|
|
|
+ minConfidence = 65 #置信度
|
|
|
|
|
|
ai_start_time = int(time.time())
|
|
|
client = boto3.client(
|
|
|
@@ -313,7 +313,7 @@ class AiView(View):
|
|
|
logger.info(labels)
|
|
|
all_labels_type = {
|
|
|
'1': ['Person', 'Human'], # 人
|
|
|
- '2': ['Dog', 'Pet', 'Canine', 'Animal', 'Puppy'], # 动物
|
|
|
+ '2': ['Dog', 'Pet', 'Canine', 'Animal', 'Puppy', 'Cat'], # 动物
|
|
|
'3': ['Car', 'Vehicle', 'Transportation', 'Automobile'] # 车
|
|
|
}
|
|
|
#找出识别的所有标签
|
|
|
@@ -321,6 +321,9 @@ class AiView(View):
|
|
|
label_name.append(label['Name'])
|
|
|
for Parents in label['Parents']:
|
|
|
label_name.append(Parents['Name'])
|
|
|
+
|
|
|
+ logger.info('标签名------')
|
|
|
+ logger.info(label_name)
|
|
|
#删除用户没有选择的ai识别类型, 并且得出最终识别结果
|
|
|
user_detect_list = user_detect_group.split(',')
|
|
|
user_detect_list = [i.strip() for i in user_detect_list]
|
