Skip to content

Commit 10bfeb2

Browse files
dizcologyleahecole
authored andcommitted
Video v1beta2 [(#1088)](#1088)
* update analyze_safe_search * update analyze_shots * update explicit_content_detection and test * update fece detection * update label detection (path) * update label detection (file) * flake * safe search --> explicit content * update faces tutorial * update client library quickstart * update shotchange tutorial * update labels tutorial * correct spelling * correction start_time_offset * import order * rebased
1 parent eb2f225 commit 10bfeb2

File tree

2 files changed

+19
-18
lines changed

2 files changed

+19
-18
lines changed

videointelligence/samples/quickstart/quickstart.py

Lines changed: 18 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,10 @@ def run_quickstart():
2626
import sys
2727
import time
2828

29-
from google.cloud.gapic.videointelligence.v1beta1 import enums
30-
from google.cloud.gapic.videointelligence.v1beta1 import (
31-
video_intelligence_service_client)
29+
from google.cloud import videointelligence_v1beta2
30+
from google.cloud.videointelligence_v1beta2 import enums
3231

33-
video_client = (video_intelligence_service_client.
34-
VideoIntelligenceServiceClient())
32+
video_client = videointelligence_v1beta2.VideoIntelligenceServiceClient()
3533
features = [enums.Feature.LABEL_DETECTION]
3634
operation = video_client.annotate_video('gs://demomaker/cat.mp4', features)
3735
print('\nProcessing video for label annotations:')
@@ -46,19 +44,22 @@ def run_quickstart():
4644
# first result is retrieved because a single video was processed
4745
results = operation.result().annotation_results[0]
4846

49-
for label in results.label_annotations:
50-
print('Label description: {}'.format(label.description))
51-
print('Locations:')
52-
53-
for l, location in enumerate(label.locations):
54-
positions = 'Entire video'
55-
if (location.segment.start_time_offset != -1 or
56-
location.segment.end_time_offset != -1):
57-
positions = '{} to {}'.format(
58-
location.segment.start_time_offset / 1000000.0,
59-
location.segment.end_time_offset / 1000000.0)
60-
print('\t{}: {}'.format(l, positions))
47+
for i, segment_label in enumerate(results.segment_label_annotations):
48+
print('Video label description: {}'.format(
49+
segment_label.entity.description))
50+
for category_entity in segment_label.category_entities:
51+
print('\tLabel category description: {}'.format(
52+
category_entity.description))
6153

54+
for i, segment in enumerate(segment_label.segments):
55+
start_time = (segment.segment.start_time_offset.seconds +
56+
segment.segment.start_time_offset.nanos / 1e9)
57+
end_time = (segment.segment.end_time_offset.seconds +
58+
segment.segment.end_time_offset.nanos / 1e9)
59+
positions = '{}s to {}s'.format(start_time, end_time)
60+
confidence = segment.confidence
61+
print('\tSegment {}: {}'.format(i, positions))
62+
print('\tConfidence: {}'.format(confidence))
6263
print('\n')
6364
# [END videointelligence_quickstart]
6465

videointelligence/samples/quickstart/quickstart_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,4 +23,4 @@
2323
def test_quickstart(capsys):
2424
quickstart.run_quickstart()
2525
out, _ = capsys.readouterr()
26-
assert 'Whiskers' in out
26+
assert 'Video label description: cat' in out

0 commit comments

Comments
 (0)