/** * @license * Copyright 2021 Google LLC. 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. * ============================================================================= */ import * as tf from '@tensorflow/tfjs-core'; import { PartWithScore } from '../types'; import { MaxHeap } from './max_heap'; /** * Builds a priority queue with part candidate positions for a specific image in * the batch. For this we find all local maxima in the score maps with score * values above a threshold. We create a single priority queue across all parts. */ export declare function buildPartWithScoreQueue(scoreThreshold: number, localMaximumRadius: number, scores: tf.TensorBuffer): MaxHeap;