AI仪表识别,服装识别

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

@@ -36,6 +36,8 @@ import com.alibaba.fastjson.JSONObject;
import lombok.extern.slf4j.Slf4j;
import javax.sound.sampled.Line;
@Slf4j
@Service
public class KafkaProcessImpl implements KafkaProcess {
@@ -129,7 +131,6 @@ public class KafkaProcessImpl implements KafkaProcess {
		}
		// 文件上传到 minio文件服务器
		multipartFile = new CommonsMultipartFile(FileUtils.createFileItem(new File(filePath), fileName));
		log.info("保存图片文件："+fileName);
		minioFileMap = fileManageService.uploadFile(multipartFile, bucketName, deviceCode, null);
@@ -375,7 +376,6 @@ public class KafkaProcessImpl implements KafkaProcess {
			log.info("开始AI人数识别处理...");
			String aiResponseData = ebcAiService.headDetect(ImageBase64Converter.convertFileToBase64(aifilePath), true,
					0.9f);
			JSONObject jsonObject = JSONObject.parseObject(aiResponseData);
			JSONArray jsonArray = jsonObject.getJSONArray("Result");
			if (!CollectionUtils.isEmpty(jsonArray)) {
@@ -384,7 +384,6 @@ public class KafkaProcessImpl implements KafkaProcess {
			} else {
				log.info("AI人数识别结果：没有人");
			}
		} catch (Exception e) {
			log.error("AI处理异常");
		}
@@ -412,12 +411,10 @@ public class KafkaProcessImpl implements KafkaProcess {
				if (resultJsonObject.getIntValue("is_match") == 0) {
					aiIdenLogDto.setIdenResult(jsonObject.getString("result"));
					aiIdenLogDto.setIdenResultType(MonitorManageConsts.AI_IDENTIFY_RESULT);
				} else {
					aiIdenLogDto.setIdenResult(jsonObject.getString("result"));
					aiIdenLogDto.setRelateEmployeeRoleId(resultJsonObject.getString("user_id"));
					aiIdenLogDto.setIdenResultType(MonitorManageConsts.AI_IDENTIFY_RESULT_ATTENDANCE);
				}
				log.info("识别成功，推送KAFKA消息");
@@ -468,8 +465,9 @@ 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消息");
                    }
@@ -496,12 +494,11 @@ private AiIdenLogDto workClothesDetect(AiIdenLogDto aiIdenLogDto, String aifileP
		List<IdenResultDto> idenResultDtoList = null;
		if(jsonObject.getString("Status").equals("Ok")){
			idenResultDtoList = new ArrayList<>();
			JSONArray jsonArray = JSONArray.parseArray(jsonObject.getString("Result"));
			if(jsonArray.size()>0){
				for (int i = 0; i < jsonArray.size(); i++) {
					JSONObject resultObj = JSONObject.parseObject(jsonArray.get(i).toString());
					JSONObject jsonArray1 = JSONObject.parseObject(resultObj.getString("clothes"));
					String aiClassName =jsonArray1.getString("class");
			List<Map> resultList = JSONArray.parseArray(jsonObject.getString("Result"),Map.class);
			if(resultList.size()>0){
				for (int i = 0; i < resultList.size(); i++) {
					JSONObject resultObject = JSONObject.parseObject(resultList.get(i).get("clothes").toString());
					String aiClassName =resultObject.getString("class");
					if(aiClassName.equals("qita")){
						idenResultDto = new IdenResultDto();
						idenResultDto.setAlarmEmployeeRoleId(MonitorManageConsts.alarmEmployeeRoleId);
@@ -526,6 +523,7 @@ private AiIdenLogDto workClothesDetect(AiIdenLogDto aiIdenLogDto, String aifileP
				}
				log.info("识别成功，推送KAFKA消息");
				processKafka(aiIdenLogDto.getTopic(), JSONObject.toJSONString(aiIdenLogDto));
				log.info("消息推送完成");
			}else {
				log.info("没有识别到人，不推送KAFKA消息");
			}
@@ -549,16 +547,17 @@ 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);
		JSONObject jsonObjectpredictions=JSONObject.parseObject(jsonObject.getString("predictions"));
		JSONObject jsonObjectpredictions=jsonObject.getJSONObject("predictions");
		if(jsonObjectpredictions.getString("Errors").isEmpty()){
			if(jsonObjectpredictions.getInteger("Nums")>0){
				log.info("识别到的仪表的数量为："+jsonObjectpredictions.getInteger("Nums"));
				aiIdenLogDto.setIdenResultType(MonitorManageConsts.AI_IDENTIFY_RESULT);
				aiIdenLogDto.setRelateEmployeeRoleId(MonitorManageConsts.alarmEmployeeRoleId);
				log.info("返回的仪表信息为："+JSON.toJSONString(jsonObject));
				aiIdenLogDto.setIdenResult(JSON.toJSONString(jsonObject));
				log.info("返回的仪表信息为："+jsonObjectpredictions.getString("Items"));
				aiIdenLogDto.setIdenResult(jsonObjectpredictions.getString("Items"));
				log.info("识别成功，推送KAFKA消息");
				processKafka(aiIdenLogDto.getTopic(), JSONObject.toJSONString(aiIdenLogDto));
				log.info("消息推送完成");
			}else{
				log.info("仪表数量为0，不推送消息");
			}
@@ -570,9 +569,6 @@ private AiIdenLogDto instrumentDetect(AiIdenLogDto aiIdenLogDto, String aifilePa
	}
	return aiIdenLogDto;
}
	/**
	 * 根据设备code获取监控任务详情
	 * @param deviceCode
@@ -580,11 +576,10 @@ private AiIdenLogDto instrumentDetect(AiIdenLogDto aiIdenLogDto, String aifilePa
	 */
	private List<Map> getResourceToolAllInfo(String deviceCode) {
		List<Map> resourceToolInfoList = new ArrayList<Map>();
		deviceCode="6L04C6FPAGFD039";
		try {
			String url = protectionResourceUrl + "?resourceToolCode=" + deviceCode;
			log.info("getResourceToolAllInfo method params: resourceToolCode=" + deviceCode + ", url=" + url);
			String resultJson = HttpServiceUtil.sendRequest(url);
			JSONObject resultData = JSON.parseObject(resultJson);
			if (StringUtils.isEmpty(resultData.getString("data"))) {