Search in sources :

Example 1 with DetectLabelsRequest

use of com.amazonaws.services.rekognition.model.DetectLabelsRequest in project myrobotlab by MyRobotLab.

the class Rekognition method getLabels.

/**
 * Hopefully, Label is serializable, otherwise it needs to return a list of
 * POJOs.
 *
 * @return
 */
public List<Label> getLabels(ByteBuffer imageBytes) {
    AmazonRekognition client = getClient();
    DetectLabelsRequest request = new DetectLabelsRequest().withImage(new Image().withBytes(imageBytes)).withMaxLabels(maxLabels).withMinConfidence(minConfidence);
    DetectLabelsResult result = client.detectLabels(request);
    List<Label> labels = result.getLabels();
    lastImage = imageBytes;
    lastLabels = labels;
    return labels;
}
Also used : AmazonRekognition(com.amazonaws.services.rekognition.AmazonRekognition) Label(com.amazonaws.services.rekognition.model.Label) DetectLabelsRequest(com.amazonaws.services.rekognition.model.DetectLabelsRequest) BufferedImage(java.awt.image.BufferedImage) Image(com.amazonaws.services.rekognition.model.Image) DetectLabelsResult(com.amazonaws.services.rekognition.model.DetectLabelsResult)

Example 2 with DetectLabelsRequest

use of com.amazonaws.services.rekognition.model.DetectLabelsRequest in project aws-doc-sdk-examples by awsdocs.

the class DetectLabels method main.

public static void main(String[] args) throws Exception {
    // Change bucket and photo to your S3 Bucket and image.
    String photo = "photo";
    String bucket = "bucket";
    AmazonRekognition rekognitionClient = AmazonRekognitionClientBuilder.defaultClient();
    DetectLabelsRequest request = new DetectLabelsRequest().withImage(new Image().withS3Object(new S3Object().withName(photo).withBucket(bucket))).withMaxLabels(10).withMinConfidence(75F);
    try {
        DetectLabelsResult result = rekognitionClient.detectLabels(request);
        List<Label> labels = result.getLabels();
        System.out.println("Detected labels for " + photo + "\n");
        for (Label label : labels) {
            System.out.println("Label: " + label.getName());
            System.out.println("Confidence: " + label.getConfidence().toString() + "\n");
            List<Instance> instances = label.getInstances();
            System.out.println("Instances of " + label.getName());
            if (instances.isEmpty()) {
                System.out.println("  " + "None");
            } else {
                for (Instance instance : instances) {
                    System.out.println("  Confidence: " + instance.getConfidence().toString());
                    System.out.println("  Bounding box: " + instance.getBoundingBox().toString());
                }
            }
            System.out.println("Parent labels for " + label.getName() + ":");
            List<Parent> parents = label.getParents();
            if (parents.isEmpty()) {
                System.out.println("  None");
            } else {
                for (Parent parent : parents) {
                    System.out.println("  " + parent.getName());
                }
            }
            System.out.println("--------------------");
            System.out.println();
        }
    } catch (AmazonRekognitionException e) {
        e.printStackTrace();
    }
}
Also used : Instance(com.amazonaws.services.rekognition.model.Instance) Parent(com.amazonaws.services.rekognition.model.Parent) Label(com.amazonaws.services.rekognition.model.Label) Image(com.amazonaws.services.rekognition.model.Image) DetectLabelsResult(com.amazonaws.services.rekognition.model.DetectLabelsResult) AmazonRekognitionException(com.amazonaws.services.rekognition.model.AmazonRekognitionException) AmazonRekognition(com.amazonaws.services.rekognition.AmazonRekognition) DetectLabelsRequest(com.amazonaws.services.rekognition.model.DetectLabelsRequest) S3Object(com.amazonaws.services.rekognition.model.S3Object)

Example 3 with DetectLabelsRequest

use of com.amazonaws.services.rekognition.model.DetectLabelsRequest in project aws-doc-sdk-examples by awsdocs.

the class DetectLabelsLocalFile method main.

public static void main(String[] args) throws Exception {
    // Change photo to the path and filename of your image.
    String photo = "input.jpg";
    ByteBuffer imageBytes;
    try (InputStream inputStream = new FileInputStream(new File(photo))) {
        imageBytes = ByteBuffer.wrap(IOUtils.toByteArray(inputStream));
    }
    AmazonRekognition rekognitionClient = AmazonRekognitionClientBuilder.defaultClient();
    DetectLabelsRequest request = new DetectLabelsRequest().withImage(new Image().withBytes(imageBytes)).withMaxLabels(10).withMinConfidence(77F);
    try {
        DetectLabelsResult result = rekognitionClient.detectLabels(request);
        List<Label> labels = result.getLabels();
        System.out.println("Detected labels for " + photo);
        for (Label label : labels) {
            System.out.println(label.getName() + ": " + label.getConfidence().toString());
        }
    } catch (AmazonRekognitionException e) {
        e.printStackTrace();
    }
}
Also used : FileInputStream(java.io.FileInputStream) InputStream(java.io.InputStream) AmazonRekognitionException(com.amazonaws.services.rekognition.model.AmazonRekognitionException) AmazonRekognition(com.amazonaws.services.rekognition.AmazonRekognition) Label(com.amazonaws.services.rekognition.model.Label) DetectLabelsRequest(com.amazonaws.services.rekognition.model.DetectLabelsRequest) Image(com.amazonaws.services.rekognition.model.Image) ByteBuffer(java.nio.ByteBuffer) File(java.io.File) DetectLabelsResult(com.amazonaws.services.rekognition.model.DetectLabelsResult) FileInputStream(java.io.FileInputStream)

Example 4 with DetectLabelsRequest

use of com.amazonaws.services.rekognition.model.DetectLabelsRequest in project amplify-android by aws-amplify.

the class AWSRekognitionService method detectLabels.

private List<Label> detectLabels(ByteBuffer imageData) throws PredictionsException {
    DetectLabelsRequest request = new DetectLabelsRequest().withImage(new Image().withBytes(imageData));
    // Detect labels in the given image via Amazon Rekognition
    final DetectLabelsResult result;
    try {
        result = rekognition.detectLabels(request);
    } catch (AmazonClientException serviceException) {
        throw new PredictionsException("Amazon Rekognition encountered an error while detecting labels.", serviceException, "See attached service exception for more details.");
    }
    List<Label> labels = new ArrayList<>();
    for (com.amazonaws.services.rekognition.model.Label rekognitionLabel : result.getLabels()) {
        List<String> parents = new ArrayList<>();
        for (Parent parent : rekognitionLabel.getParents()) {
            parents.add(parent.getName());
        }
        List<RectF> boxes = new ArrayList<>();
        for (Instance instance : rekognitionLabel.getInstances()) {
            boxes.add(RekognitionResultTransformers.fromBoundingBox(instance.getBoundingBox()));
        }
        Label amplifyLabel = Label.builder().value(rekognitionLabel.getName()).confidence(rekognitionLabel.getConfidence()).parentLabels(parents).boxes(boxes).build();
        labels.add(amplifyLabel);
    }
    return labels;
}
Also used : Parent(com.amazonaws.services.rekognition.model.Parent) Instance(com.amazonaws.services.rekognition.model.Instance) AmazonClientException(com.amazonaws.AmazonClientException) Label(com.amplifyframework.predictions.models.Label) ModerationLabel(com.amazonaws.services.rekognition.model.ModerationLabel) ArrayList(java.util.ArrayList) Image(com.amazonaws.services.rekognition.model.Image) DetectLabelsResult(com.amazonaws.services.rekognition.model.DetectLabelsResult) RectF(android.graphics.RectF) DetectLabelsRequest(com.amazonaws.services.rekognition.model.DetectLabelsRequest) PredictionsException(com.amplifyframework.predictions.PredictionsException)

Aggregations

DetectLabelsRequest (com.amazonaws.services.rekognition.model.DetectLabelsRequest)4 DetectLabelsResult (com.amazonaws.services.rekognition.model.DetectLabelsResult)4 Image (com.amazonaws.services.rekognition.model.Image)4 AmazonRekognition (com.amazonaws.services.rekognition.AmazonRekognition)3 Label (com.amazonaws.services.rekognition.model.Label)3 AmazonRekognitionException (com.amazonaws.services.rekognition.model.AmazonRekognitionException)2 Instance (com.amazonaws.services.rekognition.model.Instance)2 Parent (com.amazonaws.services.rekognition.model.Parent)2 RectF (android.graphics.RectF)1 AmazonClientException (com.amazonaws.AmazonClientException)1 ModerationLabel (com.amazonaws.services.rekognition.model.ModerationLabel)1 S3Object (com.amazonaws.services.rekognition.model.S3Object)1 PredictionsException (com.amplifyframework.predictions.PredictionsException)1 Label (com.amplifyframework.predictions.models.Label)1 BufferedImage (java.awt.image.BufferedImage)1 File (java.io.File)1 FileInputStream (java.io.FileInputStream)1 InputStream (java.io.InputStream)1 ByteBuffer (java.nio.ByteBuffer)1 ArrayList (java.util.ArrayList)1