Examples with GcsDestination com.google.cloud.documentai.v1beta2.GcsDestination used on opensource projects

Example 16 with GcsDestination

use of com.google.cloud.documentai.v1beta2.GcsDestination in project java-vision by googleapis.

the class AsyncBatchAnnotateImages method asyncBatchAnnotateImages.

public static void asyncBatchAnnotateImages(String inputImageUri, String outputUri) throws IOException, ExecutionException, InterruptedException {
    // the "close" method on the client to safely clean up any remaining background resources.
    try (ImageAnnotatorClient imageAnnotatorClient = ImageAnnotatorClient.create()) {
        // You can send multiple images to be annotated, this sample demonstrates how to do this with
        // one image. If you want to use multiple images, you have to create a `AnnotateImageRequest`
        // object for each image that you want annotated.
        // First specify where the vision api can find the image
        ImageSource source = ImageSource.newBuilder().setImageUri(inputImageUri).build();
        Image image = Image.newBuilder().setSource(source).build();
        // Set the type of annotation you want to perform on the image
        // https://cloud.google.com/vision/docs/reference/rpc/google.cloud.vision.v1#google.cloud.vision.v1.Feature.Type
        Feature feature = Feature.newBuilder().setType(Feature.Type.LABEL_DETECTION).build();
        // Build the request object for that one image. Note: for additional images you have to create
        // additional `AnnotateImageRequest` objects and store them in a list to be used below.
        AnnotateImageRequest imageRequest = AnnotateImageRequest.newBuilder().setImage(image).addFeatures(feature).build();
        // Set where to store the results for the images that will be annotated.
        GcsDestination gcsDestination = GcsDestination.newBuilder().setUri(outputUri).build();
        OutputConfig outputConfig = OutputConfig.newBuilder().setGcsDestination(gcsDestination).setBatchSize(// The max number of responses to output in each JSON file
        2).build();
        // Add each `AnnotateImageRequest` object to the batch request and add the output config.
        AsyncBatchAnnotateImagesRequest request = AsyncBatchAnnotateImagesRequest.newBuilder().addRequests(imageRequest).setOutputConfig(outputConfig).build();
        // Make the asynchronous batch request.
        AsyncBatchAnnotateImagesResponse response = imageAnnotatorClient.asyncBatchAnnotateImagesAsync(request).get();
        // The output is written to GCS with the provided output_uri as prefix
        String gcsOutputUri = response.getOutputConfig().getGcsDestination().getUri();
        System.out.format("Output written to GCS with prefix: %s%n", gcsOutputUri);
    }
}