org.apache.beam.sdk.extensions.ml

Class CloudVision

java.lang.Object

org.apache.beam.sdk.extensions.ml.CloudVision

@Experimental
public class CloudVision
extends java.lang.Object

Factory class for implementations of AnnotateImages.

Example usage:

 pipeline
  .apply(Create.of(IMAGE_URI))
  .apply(CloudVision.annotateImagesFromGcsUri(sideInputWithContext,
         features, 1, 1));
 

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	CloudVision.AnnotateImagesFromBytes Accepts ByteString (encoded image contents) with optional DoFn.SideInput with a Map of ImageContext to the image.
static class	CloudVision.AnnotateImagesFromBytesWithContext Accepts KVs of ByteString (encoded image contents) and ImageContext.
static class	CloudVision.AnnotateImagesFromGcsUri Accepts String (image URI on GCS) with optional DoFn.SideInput with a Map of ImageContext to the image.
static class	CloudVision.AnnotateImagesFromGcsUriWithContext Accepts KVs of String (GCS URI to the image) and ImageContext.

Constructor Summary

Constructors

Constructor and Description
CloudVision()

Method Summary

Modifier and Type	Method and Description
static CloudVision.AnnotateImagesFromBytes	annotateImagesFromBytes(PCollectionView<java.util.Map<ByteString,com.google.cloud.vision.v1.ImageContext>> contextSideInput, java.util.List<com.google.cloud.vision.v1.Feature> features, long batchSize) Creates a PTransform that annotates images from their contents encoded in ByteStrings.
static CloudVision.AnnotateImagesFromBytes	annotateImagesFromBytes(PCollectionView<java.util.Map<ByteString,com.google.cloud.vision.v1.ImageContext>> contextSideInput, java.util.List<com.google.cloud.vision.v1.Feature> features, long batchSize, int desiredRequestParallelism) Creates a PTransform that annotates images from their contents encoded in ByteStrings.
static CloudVision.AnnotateImagesFromBytesWithContext	annotateImagesFromBytesWithContext(java.util.List<com.google.cloud.vision.v1.Feature> features, long batchSize) Creates a PTransform that annotates images from KVs of their GCS addresses in Strings and ImageContext for each image.
static CloudVision.AnnotateImagesFromBytesWithContext	annotateImagesFromBytesWithContext(java.util.List<com.google.cloud.vision.v1.Feature> features, long batchSize, int desiredRequestParallelism) Creates a PTransform that annotates images from KVs of their GCS addresses in Strings and ImageContext for each image.
static CloudVision.AnnotateImagesFromGcsUri	annotateImagesFromGcsUri(PCollectionView<java.util.Map<java.lang.String,com.google.cloud.vision.v1.ImageContext>> contextSideInput, java.util.List<com.google.cloud.vision.v1.Feature> features, long batchSize) Creates a PTransform that annotates images from their GCS addresses.
static CloudVision.AnnotateImagesFromGcsUri	annotateImagesFromGcsUri(PCollectionView<java.util.Map<java.lang.String,com.google.cloud.vision.v1.ImageContext>> contextSideInput, java.util.List<com.google.cloud.vision.v1.Feature> features, long batchSize, int desiredRequestParallelism) Creates a PTransform that annotates images from their GCS addresses.
static CloudVision.AnnotateImagesFromGcsUriWithContext	annotateImagesFromGcsUriWithContext(java.util.List<com.google.cloud.vision.v1.Feature> features, long batchSize) Creates a PTransform that annotates images from KVs of their String-encoded contents and ImageContext for each image.
static CloudVision.AnnotateImagesFromGcsUriWithContext	annotateImagesFromGcsUriWithContext(java.util.List<com.google.cloud.vision.v1.Feature> features, long batchSize, int desiredRequestParallelism) Creates a PTransform that annotates images from KVs of their String-encoded contents and ImageContext for each image.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

CloudVision

public CloudVision()

Method Detail

annotateImagesFromGcsUri

public static CloudVision.AnnotateImagesFromGcsUri annotateImagesFromGcsUri(PCollectionView<java.util.Map<java.lang.String,com.google.cloud.vision.v1.ImageContext>> contextSideInput,
                                                                            java.util.List<com.google.cloud.vision.v1.Feature> features,
                                                                            long batchSize,
                                                                            int desiredRequestParallelism)

Creates a PTransform that annotates images from their GCS addresses.

Parameters:

contextSideInput - optional side input with contexts for select images. The ImageContext objects provide additional metadata for the annotation API. This way users can

features - annotation features that should be passed to the API

batchSize - request batch size to be sent to API. Max 16, at least 1.

desiredRequestParallelism - desired number of concurrent batched requests.

Returns:

the PTransform.

annotateImagesFromGcsUri

public static CloudVision.AnnotateImagesFromGcsUri annotateImagesFromGcsUri(PCollectionView<java.util.Map<java.lang.String,com.google.cloud.vision.v1.ImageContext>> contextSideInput,
                                                                            java.util.List<com.google.cloud.vision.v1.Feature> features,
                                                                            long batchSize)

Creates a PTransform that annotates images from their GCS addresses. Uses a default value 5 for desiredRequestParallelism.

Parameters:

contextSideInput - optional side input with contexts for select images. The ImageContext objects provide additional metadata for the annotation API. This way users can fine-tune analysis of selected images.

features - annotation features that should be passed to the API

batchSize - request batch size to be sent to API. Max 16, at least 1.

Returns:

the PTransform.

annotateImagesFromBytes

public static CloudVision.AnnotateImagesFromBytes annotateImagesFromBytes(PCollectionView<java.util.Map<ByteString,com.google.cloud.vision.v1.ImageContext>> contextSideInput,
                                                                          java.util.List<com.google.cloud.vision.v1.Feature> features,
                                                                          long batchSize,
                                                                          int desiredRequestParallelism)

Creates a PTransform that annotates images from their contents encoded in ByteStrings.

Parameters:

contextSideInput - optional side input with contexts for select images. The ImageContext objects provide additional metadata for the annotation API. This way users can fine-tune analysis of selected images.

features - annotation features that should be passed to the API

batchSize - request batch size to be sent to API. Max 16, at least 1.

desiredRequestParallelism - desired number of concurrent batched requests.

Returns:

the PTransform.

annotateImagesFromBytes

public static CloudVision.AnnotateImagesFromBytes annotateImagesFromBytes(PCollectionView<java.util.Map<ByteString,com.google.cloud.vision.v1.ImageContext>> contextSideInput,
                                                                          java.util.List<com.google.cloud.vision.v1.Feature> features,
                                                                          long batchSize)

Creates a PTransform that annotates images from their contents encoded in ByteStrings. Uses a default value 5 for desiredRequestParallelism.

Parameters:

contextSideInput - optional side input with contexts for select images. The ImageContext objects provide additional metadata for the annotation API. This way users can fine-tune analysis of selected images.

features - annotation features that should be passed to the API

batchSize - request batch size to be sent to API. Max 16, at least 1.

Returns:

the PTransform.

annotateImagesFromBytesWithContext

public static CloudVision.AnnotateImagesFromBytesWithContext annotateImagesFromBytesWithContext(java.util.List<com.google.cloud.vision.v1.Feature> features,
                                                                                                long batchSize,
                                                                                                int desiredRequestParallelism)

Creates a PTransform that annotates images from KVs of their GCS addresses in Strings and ImageContext for each image.

Parameters:

features - annotation features that should be passed to the API

batchSize - request batch size to be sent to API. Max 16, at least 1.

desiredRequestParallelism - desired number of concurrent batched requests.

Returns:

the PTransform.

annotateImagesFromBytesWithContext

public static CloudVision.AnnotateImagesFromBytesWithContext annotateImagesFromBytesWithContext(java.util.List<com.google.cloud.vision.v1.Feature> features,
                                                                                                long batchSize)

Creates a PTransform that annotates images from KVs of their GCS addresses in Strings and ImageContext for each image. Uses a default value 5 for desiredRequestParallelism.

Parameters:

features - annotation features that should be passed to the API

batchSize - request batch size to be sent to API. Max 16, at least 1.

Returns:

the PTransform.

annotateImagesFromGcsUriWithContext

public static CloudVision.AnnotateImagesFromGcsUriWithContext annotateImagesFromGcsUriWithContext(java.util.List<com.google.cloud.vision.v1.Feature> features,
                                                                                                  long batchSize,
                                                                                                  int desiredRequestParallelism)

Creates a PTransform that annotates images from KVs of their String-encoded contents and ImageContext for each image.

Parameters:

features - annotation features that should be passed to the API

batchSize - request batch size to be sent to API. Max 16, at least 1.

desiredRequestParallelism - desired number of concurrent batched requests.

Returns:

the PTransform.

annotateImagesFromGcsUriWithContext

public static CloudVision.AnnotateImagesFromGcsUriWithContext annotateImagesFromGcsUriWithContext(java.util.List<com.google.cloud.vision.v1.Feature> features,
                                                                                                  long batchSize)

Creates a PTransform that annotates images from KVs of their String-encoded contents and ImageContext for each image. Uses a default value 5 for desiredRequestParallelism.

Parameters:

features - annotation features that should be passed to the API

batchSize - request batch size to be sent to API. Max 16, at least 1.

Returns:

the PTransform.