mirror of
https://github.com/restic/restic.git
synced 2024-08-22 04:13:23 +02:00
152 lines
6.2 KiB
Go
152 lines
6.2 KiB
Go
// Copyright 2017, Google Inc. All rights reserved.
|
|
//
|
|
// 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.
|
|
|
|
package vision
|
|
|
|
import (
|
|
"github.com/googleapis/gax-go"
|
|
"golang.org/x/net/context"
|
|
pb "google.golang.org/genproto/googleapis/cloud/vision/v1"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// AnnotateImage runs image detection and annotation for a single image.
|
|
func (c *ImageAnnotatorClient) AnnotateImage(ctx context.Context, req *pb.AnnotateImageRequest, opts ...gax.CallOption) (*pb.AnnotateImageResponse, error) {
|
|
res, err := c.BatchAnnotateImages(ctx, &pb.BatchAnnotateImagesRequest{
|
|
Requests: []*pb.AnnotateImageRequest{req},
|
|
}, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.Responses[0], nil
|
|
}
|
|
|
|
// Called for a single image and a single feature.
|
|
func (c *ImageAnnotatorClient) annotateOne(ctx context.Context, img *pb.Image, ictx *pb.ImageContext, ftype pb.Feature_Type, maxResults int, opts []gax.CallOption) (*pb.AnnotateImageResponse, error) {
|
|
res, err := c.AnnotateImage(ctx, &pb.AnnotateImageRequest{
|
|
Image: img,
|
|
ImageContext: ictx,
|
|
Features: []*pb.Feature{{Type: ftype, MaxResults: int32(maxResults)}},
|
|
}, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// When there is only one image and one feature, the response's Error field is
|
|
// unambiguously about that one detection, so we "promote" it to the error return
|
|
// value.
|
|
// res.Error is a google.rpc.Status. Convert to a Go error. Use a gRPC
|
|
// error because it preserves the code as a separate field.
|
|
// TODO(jba): preserve the details field.
|
|
if res.Error != nil {
|
|
return nil, status.Errorf(codes.Code(res.Error.Code), "%s", res.Error.Message)
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
// DetectFaces performs face detection on the image.
|
|
// At most maxResults results are returned.
|
|
func (c *ImageAnnotatorClient) DetectFaces(ctx context.Context, img *pb.Image, ictx *pb.ImageContext, maxResults int, opts ...gax.CallOption) ([]*pb.FaceAnnotation, error) {
|
|
res, err := c.annotateOne(ctx, img, ictx, pb.Feature_FACE_DETECTION, maxResults, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.FaceAnnotations, nil
|
|
}
|
|
|
|
// DetectLandmarks performs landmark detection on the image.
|
|
// At most maxResults results are returned.
|
|
func (c *ImageAnnotatorClient) DetectLandmarks(ctx context.Context, img *pb.Image, ictx *pb.ImageContext, maxResults int, opts ...gax.CallOption) ([]*pb.EntityAnnotation, error) {
|
|
res, err := c.annotateOne(ctx, img, ictx, pb.Feature_LANDMARK_DETECTION, maxResults, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.LandmarkAnnotations, nil
|
|
}
|
|
|
|
// DetectLogos performs logo detection on the image.
|
|
// At most maxResults results are returned.
|
|
func (c *ImageAnnotatorClient) DetectLogos(ctx context.Context, img *pb.Image, ictx *pb.ImageContext, maxResults int, opts ...gax.CallOption) ([]*pb.EntityAnnotation, error) {
|
|
res, err := c.annotateOne(ctx, img, ictx, pb.Feature_LOGO_DETECTION, maxResults, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.LogoAnnotations, nil
|
|
}
|
|
|
|
// DetectLabels performs label detection on the image.
|
|
// At most maxResults results are returned.
|
|
func (c *ImageAnnotatorClient) DetectLabels(ctx context.Context, img *pb.Image, ictx *pb.ImageContext, maxResults int, opts ...gax.CallOption) ([]*pb.EntityAnnotation, error) {
|
|
res, err := c.annotateOne(ctx, img, ictx, pb.Feature_LABEL_DETECTION, maxResults, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.LabelAnnotations, nil
|
|
}
|
|
|
|
// DetectTexts performs text detection on the image.
|
|
// At most maxResults results are returned.
|
|
func (c *ImageAnnotatorClient) DetectTexts(ctx context.Context, img *pb.Image, ictx *pb.ImageContext, maxResults int, opts ...gax.CallOption) ([]*pb.EntityAnnotation, error) {
|
|
res, err := c.annotateOne(ctx, img, ictx, pb.Feature_TEXT_DETECTION, maxResults, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.TextAnnotations, nil
|
|
}
|
|
|
|
// DetectDocumentText performs full text (OCR) detection on the image.
|
|
func (c *ImageAnnotatorClient) DetectDocumentText(ctx context.Context, img *pb.Image, ictx *pb.ImageContext, opts ...gax.CallOption) (*pb.TextAnnotation, error) {
|
|
res, err := c.annotateOne(ctx, img, ictx, pb.Feature_DOCUMENT_TEXT_DETECTION, 0, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.FullTextAnnotation, nil
|
|
}
|
|
|
|
// DetectSafeSearch performs safe-search detection on the image.
|
|
func (c *ImageAnnotatorClient) DetectSafeSearch(ctx context.Context, img *pb.Image, ictx *pb.ImageContext, opts ...gax.CallOption) (*pb.SafeSearchAnnotation, error) {
|
|
res, err := c.annotateOne(ctx, img, ictx, pb.Feature_SAFE_SEARCH_DETECTION, 0, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.SafeSearchAnnotation, nil
|
|
}
|
|
|
|
// DetectImageProperties computes properties of the image.
|
|
func (c *ImageAnnotatorClient) DetectImageProperties(ctx context.Context, img *pb.Image, ictx *pb.ImageContext, opts ...gax.CallOption) (*pb.ImageProperties, error) {
|
|
res, err := c.annotateOne(ctx, img, ictx, pb.Feature_IMAGE_PROPERTIES, 0, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.ImagePropertiesAnnotation, nil
|
|
}
|
|
|
|
// DetectWeb computes a web annotation on the image.
|
|
func (c *ImageAnnotatorClient) DetectWeb(ctx context.Context, img *pb.Image, ictx *pb.ImageContext, opts ...gax.CallOption) (*pb.WebDetection, error) {
|
|
res, err := c.annotateOne(ctx, img, ictx, pb.Feature_WEB_DETECTION, 0, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.WebDetection, nil
|
|
}
|
|
|
|
// CropHints computes crop hints for the image.
|
|
func (c *ImageAnnotatorClient) CropHints(ctx context.Context, img *pb.Image, ictx *pb.ImageContext, opts ...gax.CallOption) (*pb.CropHintsAnnotation, error) {
|
|
res, err := c.annotateOne(ctx, img, ictx, pb.Feature_CROP_HINTS, 0, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.CropHintsAnnotation, nil
|
|
}
|