Skip to content

Commit

Permalink
Merge pull request #111 from GoogleCloudPlatform/label
Browse files Browse the repository at this point in the history
Add Cloud Vision label detection sample.
  • Loading branch information
jmdobry committed May 12, 2016
2 parents bc87638 + e3bb52c commit 1b8513e
Show file tree
Hide file tree
Showing 8 changed files with 118 additions and 2 deletions.
2 changes: 1 addition & 1 deletion test/vision/faceDetection.test.js
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ function MockCanvas () {
MockCanvas.Image = function () {};

var faceDetectionExample = require('../../vision/faceDetection');
var inputFile = path.resolve(path.join('../../vision', 'face.png'));
var inputFile = path.resolve(path.join('../../vision/resources', 'face.png'));
var outputFile = path.resolve(path.join('../../vision', 'out.png'));

test.cb('should detect faces', function (t) {
Expand Down
31 changes: 31 additions & 0 deletions test/vision/labelDetection.test.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
// Copyright 2016, Google, Inc.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

'use strict';

var test = require('ava');
var path = require('path');

var labelDetectionSample = require('../../vision/labelDetection');
var inputFile = path.resolve(path.join('../../vision/resources', 'cat.jpg'));

test.cb('should detect labels', function (t) {
labelDetectionSample.main(
inputFile,
function (err, labels) {
t.ifError(err);
t.ok(labels.length > 0);
t.end();
}
);
});
6 changes: 6 additions & 0 deletions vision/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -28,3 +28,9 @@ details.
Execute the sample:

node faceDetection "/path/to/image.jpg"

### Label detection sample

Execute the sample:

node labelDetection "/path/to/image.jpg"
78 changes: 78 additions & 0 deletions vision/labelDetection.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,78 @@
// Copyright 2016, Google, Inc.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

'use strict';

// [START app]
// [START import_libraries]
var gcloud = require('gcloud');
// [END import_libraries]

// [START authenticate]
// You must set the GOOGLE_APPLICATION_CREDENTIALS and GCLOUD_PROJECT
// environment variables to run this sample. See:
// https://github.com/GoogleCloudPlatform/gcloud-node/blob/master/docs/authentication.md
var projectId = process.env.GCLOUD_PROJECT;

// Initialize gcloud
gcloud = gcloud({
projectId: projectId
});

// Get a reference to the vision component
var vision = gcloud.vision();
// [END authenticate]

/**
* Uses the Vision API to detect labels in the given file.
*/
// [START construct_request]
function detectLabels(inputFile, callback) {
// Make a call to the Vision API to detect the labels
vision.detectLabels(inputFile, { verbose: true }, function (err, labels) {
if (err) {
return callback(err);
}
console.log('result:', JSON.stringify(labels, null, 2));
callback(null, labels);
});
}
// [END construct_request]

// Run the example
function main(inputFile, callback) {
detectLabels(inputFile, function (err, labels) {
if (err) {
return callback(err);
}

// [START parse_response]
console.log('Found label: ' + labels[0].desc + ' for ' + inputFile);
// [END parse_response]
callback(null, labels);
});
}

// [START run_application]
if (module === require.main) {
if (process.argv.length < 3) {
console.log('Usage: node labelDetection <inputFile>');
process.exit(1);
}
var inputFile = process.argv[2];
main(inputFile, console.log);
}
// [END run_application]
// [END app]

exports.main = main;
3 changes: 2 additions & 1 deletion vision/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,8 @@
"node": ">=0.10.x"
},
"scripts": {
"faceDetection": "node faceDetection.js"
"faceDetection": "node faceDetection.js",
"labelDetection": "node labelDetection.js"
},
"dependencies": {
"gcloud": "^0.32.0",
Expand Down
Binary file added vision/resources/cat.jpg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
File renamed without changes
Binary file added vision/resources/faulkner.jpg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 1b8513e

Please sign in to comment.