| | |
| | | } |
| | | InputStream inputStream = file.getInputStream(); |
| | | List<Classifications.Classification> predict = herbUtil.predict(inputStream); |
| | | Map<String, Double> collect = predict.stream().collect(Collectors.toMap(Classifications.Classification::getClassName, Classifications.Classification::getProbability)); |
| | | |
| | | if(predict.size()>0) { |
| | | Map<String, Double> collect = predict.stream().collect(Collectors.toMap(Classifications.Classification::getClassName, Classifications.Classification::getProbability)); |
| | | |
| | | |
| | | List<DryHerbInfoVo> voList = new ArrayList<>(); |
| | | Set<String> strings = collect.keySet(); |
| | | List<DryHerbInfo> list = herbInfoService.list(new LambdaQueryWrapper<DryHerbInfo>().in(DryHerbInfo::getPinyin, strings)); |
| | | list.forEach(item -> { |
| | | DryHerbInfoVo dryHerbInfoVo = new DryHerbInfoVo(); |
| | | BeanUtil.copyProperties(item, dryHerbInfoVo); |
| | | dryHerbInfoVo.setProbabily(collect.get(item.getPinyin())); |
| | | voList.add(dryHerbInfoVo); |
| | | }); |
| | | List<DryHerbInfoVo> voList = new ArrayList<>(); |
| | | Set<String> strings = collect.keySet(); |
| | | List<DryHerbInfo> list = herbInfoService.list(new LambdaQueryWrapper<DryHerbInfo>().in(DryHerbInfo::getPinyin, strings)); |
| | | list.forEach(item -> { |
| | | DryHerbInfoVo dryHerbInfoVo = new DryHerbInfoVo(); |
| | | BeanUtil.copyProperties(item, dryHerbInfoVo); |
| | | dryHerbInfoVo.setProbabily(collect.get(item.getPinyin())); |
| | | voList.add(dryHerbInfoVo); |
| | | }); |
| | | |
| | | List<DryHerbInfoVo> collect1 = voList.stream().sorted(Comparator.comparing(DryHerbInfoVo::getProbabily, Comparator.reverseOrder())). |
| | | collect(Collectors.toList()); |
| | | List<DryHerbInfoVo> collect1 = voList.stream().sorted(Comparator.comparing(DryHerbInfoVo::getProbabily, Comparator.reverseOrder())). |
| | | collect(Collectors.toList()); |
| | | |
| | | return Result.ok(collect1); |
| | | return Result.ok(collect1); |
| | | } else { |
| | | return Result.error("AI识别服务异常"); |
| | | } |
| | | |
| | | } catch (Exception e) { |
| | | e.printStackTrace(); |
| | | return Result.error("AI识别服务异常"); |