2016-09-13 127 views
1

我正在尝试向Google Vision文本检测API发出批量请求。到目前为止,我把图像的路径放入列表中,发出批量请求并获得响应。但是,我无法确定哪个结果属于哪个图像。为此,我尝试在请求中添加一个ID,当我得到结果时,我会比较ID。但是,我无法将任何自定义字段放入请求中。我的方法有什么问题吗?我怎么知道哪个响应属于哪个图像?Google Vision Java客户端批量请求响应识别

这里是我使用的这些请求的代码:

private Vision vision; 
private static final String APPLICATION_NAME = "ProjectName"; 

public static Vision getVisionService() throws IOException, GeneralSecurityException { 

    GoogleCredential credential = GoogleCredential.fromStream 
      (new FileInputStream("/project-key.json")) 
      .createScoped(VisionScopes.all()); 
    JsonFactory jsonFactory = JacksonFactory.getDefaultInstance(); 
    return new Vision.Builder(GoogleNetHttpTransport.newTrustedTransport(), jsonFactory, credential) 
      .setApplicationName(APPLICATION_NAME) 
      .build(); 
} 
/** 
* Gets up to {@code maxResults} text annotations for images stored at {@code paths}. 
*/ 
public List<String> detectText(List<Path> paths) { 
    ImmutableList.Builder<AnnotateImageRequest> requests = ImmutableList.builder(); 

    try { 
     for (Path path : paths) { 
      byte[] data; 
      data = Files.readAllBytes(path); 
      requests.add(
        new AnnotateImageRequest() 
        .setImage(new Image().encodeContent(data)) 
        .setFeatures(ImmutableList.of(
          new Feature() 
          .setType("TEXT_DETECTION") 
          .setMaxResults(10)))); 
     } 

     Vision.Images.Annotate annotate = 
       vision.images() 
       .annotate(new BatchAnnotateImagesRequest().setRequests(requests.build())); 
     // Due to a bug: requests to Vision API containing large images fail when GZipped. 
     annotate.setDisableGZipContent(true); 
     BatchAnnotateImagesResponse batchResponse = annotate.execute(); 
     assert batchResponse.getResponses().size() == paths.size(); 

     List<String> output = new ArrayList(); 

     for (int i = 0; i < paths.size(); i++) { 
      AnnotateImageResponse response = batchResponse.getResponses().get(i); 
      if(response != null && response.getTextAnnotations() != null){ 
       System.out.println(response.toString()); 
       String result = getDescriptionFromJson(response.getTextAnnotations().toString()); 
       System.out.println(response.get("customField")); 
       output.add(result); 
      } 
     } 
     return output; 
    } catch (IOException ex) { 
     System.out.println("Exception occured: " + ex); 
     return null; 
    } 
} 

public String getDescriptionFromJson(String json){ 
    JSONArray results = new JSONArray(json); 
    JSONObject result = (JSONObject) results.get(0); 
    return result.getString("description"); 
} 

public static void main(String[] args) { 
    GoogleVisionQueryHelper g = new GoogleVisionQueryHelper(); 

    try { 
     g.vision = getVisionService(); 
     List<Path> paths = new ArrayList<>(); 

     String directory = "/images"; 

     File[] files = new File(directory).listFiles(); 
     for(File file : files){ 
      if(file.isFile() && !file.getName().contains("DS_Store") && !file.getName().startsWith(".")){ 
      System.out.println(file.getAbsolutePath()); 
      paths.add(Paths.get(file.getAbsolutePath())); 
      } 
     }   
     System.out.println("Starting..."); 

     for(String s: g.detectText(paths)){ 
      System.out.println(s); 
     } 
    } catch (IOException | GeneralSecurityException e) { 
     // TODO Auto-generated catch block 
     e.printStackTrace(); 
    } 
} 
+0

嗨,兄弟,你能给我一些帮助吗?我的计划将停留在 'BatchAnnotateImagesResponse响应= vision.images() .annotate(requestBatch) .setDisableGZipContent(真) .execute();' 你知道原因吗?我对Google云视觉api完全陌生 我怀疑请求网址的正确性,因为它总是显示“https:// vision.googleapis.com”,而不是“https://vision.googleapis.com/v1”/images:annotate?key =' –

回答

1

BatchAnnotateImageResponse持有相同的顺序在BatchAnnotateImageRequest请求列表的响应列表。

+0

你怎么知道的? – user2604150