Жадно выбирает подмножество ограничивающих рамок в порядке убывания количества очков,
Эта операция выполняет non_max_suppression для входных данных каждого пакета во всех классах. Удаляет блоки с высоким уровнем пересечения-объединения (IOU), перекрывающиеся с ранее выбранными блоками. Ограничительные рамки предоставляются в виде [y1, x1, y2, x2], где (y1, x1) и (y2, x2) — координаты любой диагональной пары углов прямоугольника, а координаты могут быть предоставлены как нормализованные (т. е. лежащие в интервал [0, 1]) или абсолютный. Обратите внимание, что этот алгоритм не зависит от того, где находится начало координат в системе координат. Также обратите внимание, что этот алгоритм инвариантен к ортогональным преобразованиям и переводам системы координат; таким образом, перевод или отражение системы координат приводит к тому, что алгоритм выбирает одни и те же поля. Результатом этой операции является окончательный тензор блоков, оценок и классов, возвращаемый после выполнения non_max_suppression.
Вложенные классы
| сорт | КомбинированныйNonMaxSuppression.Options | Дополнительные атрибуты для CombinedNonMaxSuppression . | |
Публичные методы
| статический CombinedNonMaxSuppression.Options | clipBoxes (логические клипбоксы) |
| статический КомбинированныйNonMaxSuppression | create ( Область области, поля операнда <Float>, оценки операнда <Float>, операнд <Integer> maxOutputSizePerClass, операнд <Integer> maxTotalSize, операнд <Float> iouThreshold, операнд <Float> ScoreThreshold, параметры... параметры) Фабричный метод для создания класса, обертывающего новую операцию CombindNonMaxSuppression. |
| Выход <С плавающей запятой> | нмседабоксес () Тензор [batch_size, max_detections, 4] float32, содержащий немаксимальные подавленные поля. |
| Выход <С плавающей запятой> | нмседаклассес () Тензор [batch_size, max_detections] float32, содержащий классы для блоков. |
| Выход <С плавающей запятой> | нмседСкорс () Тензор [batch_size, max_detections] float32, содержащий оценки для ящиков. |
| статический CombinedNonMaxSuppression.Options | PadPerClass (логическое значение PadPerClass) |
| Вывод <Целое число> | действительные обнаружения () Тензор [batch_size] int32, указывающий количество действительных обнаружений на элемент пакета. |
Унаследованные методы
Публичные методы
общедоступные статические CombindNonMaxSuppression.Options clipBoxes (логические clipBoxes)
Параметры
| клипбоксы | Если это правда, предположим, что координаты поля находятся между [0, 1] и обрежем выходные поля, если они выходят за пределы [0, 1]. Если false, не выполнять обрезку и выводить координаты поля как есть. |
|---|
public static JointNonMaxSuppression create (область области действия , поля Operand <Float>, оценки Operand <Float>, Operand <Integer> maxOutputSizePerClass, Operand <Integer> maxTotalSize, Operand <Float> iouThreshold, Operand <Float> ScoreThreshold, Options... options)
Фабричный метод для создания класса, обертывающего новую операцию CombindNonMaxSuppression.
Параметры
| объем | текущий объем |
|---|---|
| коробки | Четырехмерный тензор с плавающей запятой формы `[batch_size, num_boxes, q, 4]`. Если `q` равно 1, то для всех классов используются одни и те же поля. В противном случае, если `q` равно количеству классов, используются поля, специфичные для класса. |
| баллы | Трехмерный тензор с плавающей запятой формы `[batch_size, num_boxes, num_classes]`, представляющий одну оценку, соответствующую каждому блоку (каждому ряду блоков). |
| Максаутпутсизеперкласс | Скалярный целочисленный тензор, представляющий максимальное количество блоков, которые должны быть выбраны с помощью немаксимального подавления для каждого класса. |
| Макстоталсизе | Скаляр int32, представляющий максимальное количество блоков, сохраняемых во всех классах. Обратите внимание, что установка большого значения для этого значения может привести к ошибке OOM в зависимости от рабочей нагрузки системы. |
| IouThreshold | Тензор с плавающей запятой 0-D, представляющий порог для принятия решения о том, не слишком ли сильно перекрываются блоки по отношению к IOU. |
| ОценкаПорог | Тензор с плавающей запятой 0D, представляющий порог для принятия решения об удалении блоков на основе оценки. |
| параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр JointNonMaxSuppression
общедоступный вывод <Float> nmsedBoxes ()
Тензор [batch_size, max_detections, 4] float32, содержащий немаксимальные подавленные поля.
общедоступный вывод <Float> nmsedClasses ()
Тензор [batch_size, max_detections] float32, содержащий классы для блоков.
общедоступный вывод <Float> nmsedScores ()
Тензор [batch_size, max_detections] float32, содержащий оценки для ящиков.
общедоступный статический КомбинированныйNonMaxSuppression.Options PadPerClass (логический PadPerClass)
Параметры
| PadPerClass | Если значение равно false, выходные поля, оценки и классы nmsed дополняются/обрезаются до `max_total_size`. Если это правда, выходные поля, оценки и классы nmsed дополняются до длины `max_size_per_class`*`num_classes`, если только она не превышает `max_total_size`, и в этом случае она обрезается до `max_total_size`. По умолчанию ложь. |
|---|
общедоступный вывод <Integer> validDetections ()
Тензор [batch_size] int32, указывающий количество действительных обнаружений на элемент пакета. Действительны только верхние записи num_detections[i] в nms_boxes[i], nms_scores[i] и nms_class[i]. Остальные записи представляют собой дополнения нулями.