AI识别

monitor-manage-service/src/main/java/com/ai/bss/monitorManage/service/impl/KafkaProcessImpl.java

@@ -292,13 +292,13 @@ public class KafkaProcessImpl implements KafkaProcess {
	 */
	private AiIdenLogDto aiHelmetDetect(AiIdenLogDto aiIdenLogDto, String aifilePath) {
		try {
			log.info("开始AI处理...");
			log.info("开始AI帽子识别处理...");
			// 安全帽识别
			String aiResponseData = ebcAiService.helmetDetect(ImageBase64Converter.convertFileToBase64(aifilePath),
					Strings.EMPTY, Strings.EMPTY);
//			String aiResponseData = ebcAiService.helmetDetect(ImageBase64Converter.convertInputStreamToBase64(multipartFile.getInputStream()), Strings.EMPTY, Strings.EMPTY);
			JSONObject jsonObject = JSONObject.parseObject(aiResponseData);// JSONUtil.formatJsonStr(aiResponseData);
			System.out.println("AI 识别结果： " + jsonObject);
			System.out.println("AI帽子识别结果： " + jsonObject);
			IdenResultDto idenResultDto = null;
			List<IdenResultDto> idenResultDtoList = null;
			if (jsonObject.getString("Status").equals("Ok")) {
@@ -343,7 +343,7 @@ public class KafkaProcessImpl implements KafkaProcess {
			// AI模型
//			aiIdenLogDto.setAiIdenModel(MonitorManageConsts.AI_MODEL_CLOTHING_CODE);
		} catch (Exception e) {
			log.error("AI处理异常");
			log.error("AI处理异常，帽子识别");
		}
		return aiIdenLogDto;
	}
@@ -385,7 +385,7 @@ public class KafkaProcessImpl implements KafkaProcess {
				log.info("AI人数识别结果：没有人");
			}
		} catch (Exception e) {
			log.error("AI处理异常");
			log.error("AI处理异常，人数识别");
		}
		return isHuman;
	}
@@ -424,7 +424,7 @@ public class KafkaProcessImpl implements KafkaProcess {
				return aiIdenLogDto;
			}
		} catch (Exception e) {
			log.error("AI处理异常");
			log.error("AI处理异常，人脸识别");
		}
		return aiIdenLogDto;
	}
@@ -465,17 +465,17 @@ public class KafkaProcessImpl implements KafkaProcess {
                                    aiIdenLogDto.setIdenResultType(MonitorManageConsts.AI_IDENTIFY_RESULT);
                                }
                        }
						log.info("识别成功，推送KAFKA消息");
						log.info("吸烟识别成功，推送KAFKA消息");
                        processKafka(aiIdenLogDto.getTopic(), JSONObject.toJSONString(aiIdenLogDto));
                    	log.info("消息推送完成");
                    }else{
                        log.info("没有识别到吸烟者，不推送KAFKA消息");
                    }
            }else{
            	log.error("识别异常返回");
            	log.error("吸烟识别识别异常");
			}
        }catch(Exception e){
        log.error("AI处理异常");
        log.error("AI处理异常，吸烟识别");
        }
        return aiIdenLogDto;
    }
@@ -489,7 +489,7 @@ private AiIdenLogDto workClothesDetect(AiIdenLogDto aiIdenLogDto, String aifileP
		//工作服识别
		String aiResponseData=ebcAiService.workClothesDetect(ImageBase64Converter.convertFileToBase64(aifilePath),true,true,0.9f);
		JSONObject jsonObject = JSONObject.parseObject(aiResponseData);
		System.out.print("AI识别结果："+aiResponseData);
		System.out.print("AI工作服识别结果："+aiResponseData);
		IdenResultDto idenResultDto = null;
		List<IdenResultDto> idenResultDtoList = null;
		if(jsonObject.getString("Status").equals("Ok")){
@@ -521,18 +521,18 @@ private AiIdenLogDto workClothesDetect(AiIdenLogDto aiIdenLogDto, String aifileP
						log.info("识别到工作服,工作服类型为："+aiClassName);
					}
				}
				log.info("识别成功，推送KAFKA消息");
				log.info("工作服识别成功，推送KAFKA消息");
				processKafka(aiIdenLogDto.getTopic(), JSONObject.toJSONString(aiIdenLogDto));
				log.info("消息推送完成");
			}else {
				log.info("没有识别到人，不推送KAFKA消息");
			}
		}else if(jsonObject.getString("Status").equals("ERROR")){
			log.error("异常信息为："+jsonObject.getString("Des"));
			log.error("工作服识别异常："+jsonObject.getString("Des"));
		}
	}catch(Exception e){
	log.error("AI处理异常");
	log.error("AI处理异常，工作服识别");
	}
	return aiIdenLogDto;
}
@@ -546,7 +546,7 @@ private AiIdenLogDto instrumentDetect(AiIdenLogDto aiIdenLogDto, String aifilePa
		//仪表识别
		String aiResponseData=ebcAiService.instrumentDetect(ImageBase64Converter.convertFileToBase64(aifilePath),Strings.EMPTY,Strings.EMPTY,Strings.EMPTY);
		JSONObject jsonObject = JSONObject.parseObject(aiResponseData);
		System.out.print("AI识别结果："+aiResponseData);
		System.out.print("AI仪表识别结果："+aiResponseData);
		JSONObject jsonObjectpredictions=jsonObject.getJSONObject("predictions");
		if(jsonObjectpredictions.getString("Errors").isEmpty()){
			if(jsonObjectpredictions.getInteger("Nums")>0){
@@ -555,7 +555,7 @@ private AiIdenLogDto instrumentDetect(AiIdenLogDto aiIdenLogDto, String aifilePa
				aiIdenLogDto.setRelateEmployeeRoleId(MonitorManageConsts.alarmEmployeeRoleId);
				log.info("返回的仪表信息为："+jsonObjectpredictions.getString("Items"));
				aiIdenLogDto.setIdenResult(jsonObjectpredictions.getString("Items"));
				log.info("识别成功，推送KAFKA消息");
				log.info("仪表识别成功，推送KAFKA消息");
				processKafka(aiIdenLogDto.getTopic(), JSONObject.toJSONString(aiIdenLogDto));
				log.info("消息推送完成");
			}else{
@@ -565,7 +565,7 @@ private AiIdenLogDto instrumentDetect(AiIdenLogDto aiIdenLogDto, String aifilePa
			log.info("异常返回，异常原因为:"+jsonObjectpredictions.getString("Errors"));
		}
	}catch(Exception e){
		log.error("AI处理异常");
		log.error("AI处理异常，仪表识别");
	}
	return aiIdenLogDto;
}