Skip to content

video: adding GA samples for logo detection #3098

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 14 commits into from
Mar 20, 2020
Merged

video: adding GA samples for logo detection #3098

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
7e5cf15
DO NOT MERGE. vision: adding GA samples for logo detection
czahedi Mar 12, 2020
c2c5418
fixing linter issues across several sample files
czahedi Mar 12, 2020
220264d
responding to Noah's comments
czahedi Mar 13, 2020
433bd76
Merge branch 'master' into vision-logo-v1-attempt2
czahedi Mar 16, 2020
adc86e6
Merge branch 'master' into vision-logo-v1-attempt2
czahedi Mar 17, 2020
e2ad6c5
responding to comments - fixing whitespace and file references
czahedi Mar 18, 2020
a594511
Merge branch 'vision-logo-v1-attempt2' of https://github.com/GoogleCl…
czahedi Mar 18, 2020
eae0f4a
Merge branch 'master' into vision-logo-v1-attempt2
czahedi Mar 18, 2020
b71eae2
Addressing comments
czahedi Mar 18, 2020
25cb250
Update video/cloud-client/analyze/video_detect_logo_gcs.py
czahedi Mar 19, 2020
b96be92
Merge branch 'master' into vision-logo-v1-attempt2
czahedi Mar 20, 2020
527c67f
fixing broken test
czahedi Mar 20, 2020
6bb5662
Merge branch 'vision-logo-v1-attempt2' of https://github.com/GoogleCl…
czahedi Mar 20, 2020
3a3c88c
typo
czahedi Mar 20, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
443 changes: 243 additions & 200 deletions video/cloud-client/analyze/analyze.py
View file
Open in desktop

Large diffs are not rendered by default.

41 changes: 25 additions & 16 deletions video/cloud-client/analyze/analyze_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,43 +18,53 @@

import analyze

POSSIBLE_TEXTS = ['Google', 'SUR', 'SUR', 'ROTO', 'Vice President', '58oo9',
'LONDRES', 'OMAR', 'PARIS', 'METRO', 'RUE', 'CARLO']
POSSIBLE_TEXTS = [
"Google",
"SUR",
"SUR",
"ROTO",
"Vice President",
"58oo9",
"LONDRES",
"OMAR",
"PARIS",
"METRO",
"RUE",
"CARLO",
]


@pytest.mark.slow
def test_analyze_shots(capsys):
analyze.analyze_shots('gs://cloud-samples-data/video/gbikes_dinosaur.mp4')
analyze.analyze_shots("gs://cloud-samples-data/video/gbikes_dinosaur.mp4")
out, _ = capsys.readouterr()
assert 'Shot 1:' in out
assert "Shot 1:" in out


@pytest.mark.slow
def test_analyze_labels(capsys):
analyze.analyze_labels('gs://cloud-samples-data/video/cat.mp4')
analyze.analyze_labels("gs://cloud-samples-data/video/cat.mp4")
out, _ = capsys.readouterr()
assert 'label description: cat' in out
assert "label description: cat" in out


@pytest.mark.slow
def test_analyze_explicit_content(capsys):
analyze.analyze_explicit_content('gs://cloud-samples-data/video/cat.mp4')
analyze.analyze_explicit_content("gs://cloud-samples-data/video/cat.mp4")
out, _ = capsys.readouterr()
assert 'pornography' in out
assert "pornography" in out


@pytest.mark.slow
def test_speech_transcription(capsys):
analyze.speech_transcription(
'gs://cloud-samples-data/video/googlework_short.mp4')
analyze.speech_transcription("gs://cloud-samples-data/video/googlework_short.mp4")
out, _ = capsys.readouterr()
assert 'cultural' in out
assert "cultural" in out


@pytest.mark.slow
def test_detect_text_gcs(capsys):
analyze.video_detect_text_gcs(
'gs://cloud-samples-data/video/googlework_tiny.mp4')
analyze.video_detect_text_gcs("gs://cloud-samples-data/video/googlework_tiny.mp4")
out, _ = capsys.readouterr()

text_exists = False
Expand All @@ -67,7 +77,6 @@ def test_detect_text_gcs(capsys):

@pytest.mark.slow
def test_track_objects_gcs(capsys):
analyze.track_objects_gcs(
'gs://cloud-samples-data/video/cat.mp4')
analyze.track_objects_gcs("gs://cloud-samples-data/video/cat.mp4")
out, _ = capsys.readouterr()
assert 'cat' in out
assert "cat" in out
Loading