Get hands-on experience with 20+ free Google Cloud products and $300 in free credit for new customers.

Bug with Max Predictions in AutoML Batch Predictions (image object detection)

I am not able to get the batch prediction API for an AutoML image object detection request to recognize a "maxPrediction" parameter. I am running the prediction against an image that contains several hundred objects.

The response I am getting back is showing 100 object detections regardless of what I specify as the  "maxPredictions" parameter.

I am using the exact "curl" approach specified here, where the "maxPredictions" parameter is mentioned in the API method:
https://cloud.google.com/vertex-ai/docs/image-data/classification/get-predictions?authuser=0#api_2

Background

The documentation here specifies that the default number of bounding boxes returned from a prediction request is 100, and the maximum is 500:
https://cloud.google.com/vertex-ai/docs/image-data/object-detection/prepare-data?authuser=0

The ImageObjectDetection definition shows support for a "maxPredictions" parameter:
gs://google-cloud-aiplatform/schema/predict/params/image_object_detection_1.0.0.yaml

1 1 166
1 REPLY 1