Search in sources :

Example 1 with LabelFrame

use of com.google.cloud.videointelligence.v1p3beta1.LabelFrame in project java-video-intelligence by googleapis.

the class StreamingAutoMlClassification method streamingAutoMlClassification.

// Perform streaming video classification with an AutoML Model
static void streamingAutoMlClassification(String filePath, String projectId, String modelId) throws TimeoutException, StatusRuntimeException, IOException {
    try (StreamingVideoIntelligenceServiceClient client = StreamingVideoIntelligenceServiceClient.create()) {
        Path path = Paths.get(filePath);
        byte[] data = Files.readAllBytes(path);
        // Set the chunk size to 5MB (recommended less than 10MB).
        int chunkSize = 5 * 1024 * 1024;
        int numChunks = (int) Math.ceil((double) data.length / chunkSize);
        String modelPath = String.format("projects/%s/locations/us-central1/models/%s", projectId, modelId);
        System.out.println(modelPath);
        StreamingAutomlClassificationConfig streamingAutomlClassificationConfig = StreamingAutomlClassificationConfig.newBuilder().setModelName(modelPath).build();
        StreamingVideoConfig streamingVideoConfig = StreamingVideoConfig.newBuilder().setFeature(StreamingFeature.STREAMING_AUTOML_CLASSIFICATION).setAutomlClassificationConfig(streamingAutomlClassificationConfig).build();
        BidiStream<StreamingAnnotateVideoRequest, StreamingAnnotateVideoResponse> call = client.streamingAnnotateVideoCallable().call();
        // The first request must **only** contain the audio configuration:
        call.send(StreamingAnnotateVideoRequest.newBuilder().setVideoConfig(streamingVideoConfig).build());
        // Send the requests in chunks
        for (int i = 0; i < numChunks; i++) {
            call.send(StreamingAnnotateVideoRequest.newBuilder().setInputContent(ByteString.copyFrom(Arrays.copyOfRange(data, i * chunkSize, i * chunkSize + chunkSize))).build());
        }
        // Tell the service you are done sending data
        call.closeSend();
        for (StreamingAnnotateVideoResponse response : call) {
            if (response.hasError()) {
                System.out.println(response.getError().getMessage());
                break;
            }
            StreamingVideoAnnotationResults annotationResults = response.getAnnotationResults();
            for (LabelAnnotation annotation : annotationResults.getLabelAnnotationsList()) {
                String entity = annotation.getEntity().getDescription();
                // There is only one frame per annotation
                LabelFrame labelFrame = annotation.getFrames(0);
                double offset = labelFrame.getTimeOffset().getSeconds() + labelFrame.getTimeOffset().getNanos() / 1e9;
                float confidence = labelFrame.getConfidence();
                System.out.format("At %fs segment: %s (%f)\n", offset, entity, confidence);
            }
        }
        System.out.println("Video streamed successfully.");
    }
}
Also used : Path(java.nio.file.Path) StreamingVideoIntelligenceServiceClient(com.google.cloud.videointelligence.v1p3beta1.StreamingVideoIntelligenceServiceClient) StreamingVideoAnnotationResults(com.google.cloud.videointelligence.v1p3beta1.StreamingVideoAnnotationResults) StreamingAutomlClassificationConfig(com.google.cloud.videointelligence.v1p3beta1.StreamingAutomlClassificationConfig) LabelFrame(com.google.cloud.videointelligence.v1p3beta1.LabelFrame) StreamingAnnotateVideoRequest(com.google.cloud.videointelligence.v1p3beta1.StreamingAnnotateVideoRequest) ByteString(com.google.protobuf.ByteString) LabelAnnotation(com.google.cloud.videointelligence.v1p3beta1.LabelAnnotation) StreamingVideoConfig(com.google.cloud.videointelligence.v1p3beta1.StreamingVideoConfig) StreamingAnnotateVideoResponse(com.google.cloud.videointelligence.v1p3beta1.StreamingAnnotateVideoResponse)

Example 2 with LabelFrame

use of com.google.cloud.videointelligence.v1p3beta1.LabelFrame in project java-video-intelligence by googleapis.

the class StreamingLabelDetection method streamingLabelDetection.

// Perform streaming video label detection
static void streamingLabelDetection(String filePath) throws IOException, TimeoutException, StatusRuntimeException {
    try (StreamingVideoIntelligenceServiceClient client = StreamingVideoIntelligenceServiceClient.create()) {
        Path path = Paths.get(filePath);
        byte[] data = Files.readAllBytes(path);
        // Set the chunk size to 5MB (recommended less than 10MB).
        int chunkSize = 5 * 1024 * 1024;
        int numChunks = (int) Math.ceil((double) data.length / chunkSize);
        StreamingLabelDetectionConfig labelConfig = StreamingLabelDetectionConfig.newBuilder().setStationaryCamera(false).build();
        StreamingVideoConfig streamingVideoConfig = StreamingVideoConfig.newBuilder().setFeature(StreamingFeature.STREAMING_LABEL_DETECTION).setLabelDetectionConfig(labelConfig).build();
        BidiStream<StreamingAnnotateVideoRequest, StreamingAnnotateVideoResponse> call = client.streamingAnnotateVideoCallable().call();
        // The first request must **only** contain the audio configuration:
        call.send(StreamingAnnotateVideoRequest.newBuilder().setVideoConfig(streamingVideoConfig).build());
        // Send the requests in chunks
        for (int i = 0; i < numChunks; i++) {
            call.send(StreamingAnnotateVideoRequest.newBuilder().setInputContent(ByteString.copyFrom(Arrays.copyOfRange(data, i * chunkSize, i * chunkSize + chunkSize))).build());
        }
        // Tell the service you are done sending data
        call.closeSend();
        for (StreamingAnnotateVideoResponse response : call) {
            StreamingVideoAnnotationResults annotationResults = response.getAnnotationResults();
            for (LabelAnnotation annotation : annotationResults.getLabelAnnotationsList()) {
                String entity = annotation.getEntity().getDescription();
                // There is only one frame per annotation
                LabelFrame labelFrame = annotation.getFrames(0);
                double offset = labelFrame.getTimeOffset().getSeconds() + labelFrame.getTimeOffset().getNanos() / 1e9;
                float confidence = labelFrame.getConfidence();
                System.out.format("%fs: %s (%f)\n", offset, entity, confidence);
            }
        }
    }
}
Also used : Path(java.nio.file.Path) StreamingVideoIntelligenceServiceClient(com.google.cloud.videointelligence.v1p3beta1.StreamingVideoIntelligenceServiceClient) StreamingVideoAnnotationResults(com.google.cloud.videointelligence.v1p3beta1.StreamingVideoAnnotationResults) LabelFrame(com.google.cloud.videointelligence.v1p3beta1.LabelFrame) StreamingAnnotateVideoRequest(com.google.cloud.videointelligence.v1p3beta1.StreamingAnnotateVideoRequest) LabelAnnotation(com.google.cloud.videointelligence.v1p3beta1.LabelAnnotation) ByteString(com.google.protobuf.ByteString) StreamingVideoConfig(com.google.cloud.videointelligence.v1p3beta1.StreamingVideoConfig) StreamingAnnotateVideoResponse(com.google.cloud.videointelligence.v1p3beta1.StreamingAnnotateVideoResponse) StreamingLabelDetectionConfig(com.google.cloud.videointelligence.v1p3beta1.StreamingLabelDetectionConfig)

Example 3 with LabelFrame

use of com.google.cloud.videointelligence.v1p3beta1.LabelFrame in project java-video-intelligence by googleapis.

the class StreamingAutoMlActionRecognition method streamingAutoMlActionRecognition.

// Perform streaming video action recognition
static void streamingAutoMlActionRecognition(String filePath, String projectId, String modelId) throws IOException, TimeoutException, StatusRuntimeException {
    try (StreamingVideoIntelligenceServiceClient client = StreamingVideoIntelligenceServiceClient.create()) {
        Path path = Paths.get(filePath);
        byte[] data = Files.readAllBytes(path);
        // Set the chunk size to 5MB (recommended less than 10MB).
        int chunkSize = 5 * 1024 * 1024;
        int numChunks = (int) Math.ceil((double) data.length / chunkSize);
        String modelPath = String.format("projects/%s/locations/us-central1/models/%s", projectId, modelId);
        System.out.println(modelPath);
        StreamingAutomlActionRecognitionConfig streamingAutomlActionRecognitionConfig = StreamingAutomlActionRecognitionConfig.newBuilder().setModelName(modelPath).build();
        StreamingVideoConfig streamingVideoConfig = StreamingVideoConfig.newBuilder().setFeature(StreamingFeature.STREAMING_AUTOML_ACTION_RECOGNITION).setAutomlActionRecognitionConfig(streamingAutomlActionRecognitionConfig).build();
        BidiStream<StreamingAnnotateVideoRequest, StreamingAnnotateVideoResponse> call = client.streamingAnnotateVideoCallable().call();
        // The first request must **only** contain the video configuration:
        call.send(StreamingAnnotateVideoRequest.newBuilder().setVideoConfig(streamingVideoConfig).build());
        // Send the requests in chunks
        for (int i = 0; i < numChunks; i++) {
            call.send(StreamingAnnotateVideoRequest.newBuilder().setInputContent(ByteString.copyFrom(Arrays.copyOfRange(data, i * chunkSize, i * chunkSize + chunkSize))).build());
        }
        // Tell the service you are done sending data
        call.closeSend();
        for (StreamingAnnotateVideoResponse response : call) {
            if (response.hasError()) {
                System.out.println(response.getError().getMessage());
                break;
            }
            StreamingVideoAnnotationResults annotationResults = response.getAnnotationResults();
            for (LabelAnnotation annotation : annotationResults.getLabelAnnotationsList()) {
                String entity = annotation.getEntity().getDescription();
                // There is only one frame per annotation
                LabelFrame labelFrame = annotation.getFrames(0);
                double offset = labelFrame.getTimeOffset().getSeconds() + labelFrame.getTimeOffset().getNanos() / 1e9;
                float confidence = labelFrame.getConfidence();
                System.out.format("At %fs segment: %s (%f)\n", offset, entity, confidence);
            }
        }
        System.out.println("Video streamed successfully.");
    }
}
Also used : Path(java.nio.file.Path) StreamingVideoIntelligenceServiceClient(com.google.cloud.videointelligence.v1p3beta1.StreamingVideoIntelligenceServiceClient) StreamingVideoAnnotationResults(com.google.cloud.videointelligence.v1p3beta1.StreamingVideoAnnotationResults) LabelFrame(com.google.cloud.videointelligence.v1p3beta1.LabelFrame) StreamingAnnotateVideoRequest(com.google.cloud.videointelligence.v1p3beta1.StreamingAnnotateVideoRequest) ByteString(com.google.protobuf.ByteString) LabelAnnotation(com.google.cloud.videointelligence.v1p3beta1.LabelAnnotation) StreamingAutomlActionRecognitionConfig(com.google.cloud.videointelligence.v1p3beta1.StreamingAutomlActionRecognitionConfig) StreamingVideoConfig(com.google.cloud.videointelligence.v1p3beta1.StreamingVideoConfig) StreamingAnnotateVideoResponse(com.google.cloud.videointelligence.v1p3beta1.StreamingAnnotateVideoResponse)

Aggregations

LabelAnnotation (com.google.cloud.videointelligence.v1p3beta1.LabelAnnotation)3 LabelFrame (com.google.cloud.videointelligence.v1p3beta1.LabelFrame)3 StreamingAnnotateVideoRequest (com.google.cloud.videointelligence.v1p3beta1.StreamingAnnotateVideoRequest)3 StreamingAnnotateVideoResponse (com.google.cloud.videointelligence.v1p3beta1.StreamingAnnotateVideoResponse)3 StreamingVideoAnnotationResults (com.google.cloud.videointelligence.v1p3beta1.StreamingVideoAnnotationResults)3 StreamingVideoConfig (com.google.cloud.videointelligence.v1p3beta1.StreamingVideoConfig)3 StreamingVideoIntelligenceServiceClient (com.google.cloud.videointelligence.v1p3beta1.StreamingVideoIntelligenceServiceClient)3 ByteString (com.google.protobuf.ByteString)3 Path (java.nio.file.Path)3 StreamingAutomlActionRecognitionConfig (com.google.cloud.videointelligence.v1p3beta1.StreamingAutomlActionRecognitionConfig)1 StreamingAutomlClassificationConfig (com.google.cloud.videointelligence.v1p3beta1.StreamingAutomlClassificationConfig)1 StreamingLabelDetectionConfig (com.google.cloud.videointelligence.v1p3beta1.StreamingLabelDetectionConfig)1