test_nms.cpp File Reference

#include "pcod_common/nms.hpp"
#include <cassert>
#include <cmath>
#include <vector>

Go to the source code of this file.

Functions
int	main ()

Function Documentation

main()

int main

(

)

Run rotated NMS regression checks.

Definition at line 61 of file test_nms.cpp.

           {
  {
    pcod_common::BoundingBox box_a;
    box_a.center = {0.0f, 0.0f};
    box_a.length = 4.0f;
    box_a.width = 2.0f;
    box_a.existence_probability = 0.9f;
    box_a.classification.push_back({0, 1.0f});
 
    pcod_common::BoundingBox box_b = box_a;
    box_b.center = {0.5f, 0.0f};
    box_b.existence_probability = 0.8f;
 
    std::vector<pcod_common::BoundingBox> boxes{box_a, box_b};
 
    pcod_common::NmsConfig config;
    config.score_thresholds = {0.1f};
    config.iou_threshold = 0.1f;
    config.max_detections = 10;
 
    pcod_common::ApplyRotatedNms(boxes, config);
    assert(boxes.size() == 1);
    assert(std::abs(boxes[0].center[0]) < 1e-6f);
    assert(std::abs(boxes[0].center[1]) < 1e-6f);
  }
 
  {
    RunNmsCase(
        {
            MakeBox(0, 0.0f, 0.0f, 4.0f, 2.0f, 0.0f, 0.9f),
            MakeBox(1, 0.0f, 0.0f, 4.0f, 2.0f, 0.0f, 0.8f),
        },
        0.1f, 10, {0});
 
    RunNmsCase(
        {
            MakeBox(0, 0.0f, 0.0f, 4.0f, 2.0f, 0.0f, 0.9f),
            MakeBox(1, 0.5f, 0.0f, 4.0f, 2.0f, 0.0f, 0.8f),
        },
        0.1f, 10, {0});
 
    RunNmsCase(
        {
            MakeBox(0, 0.0f, 0.0f, 4.0f, 2.0f, 0.0f, 0.9f),
            MakeBox(1, 3.5f, 0.0f, 4.0f, 2.0f, 0.0f, 0.8f),
        },
        0.1f, 10, {0, 1});
 
    RunNmsCase(
        {
            MakeBox(0, 0.0f, 0.0f, 4.0f, 2.0f, 0.0f, 0.9f),
            MakeBox(1, 4.0f, 0.0f, 4.0f, 2.0f, 0.0f, 0.8f),
        },
        0.0f, 10, {0, 1});
 
    RunNmsCase(
        {
            MakeBox(0, 0.0f, 0.0f, 6.0f, 4.0f, 0.0f, 0.9f),
            MakeBox(1, 0.0f, 0.0f, 2.0f, 1.0f, 0.0f, 0.8f),
        },
        0.05f, 10, {0});
 
    RunNmsCase(
        {
            MakeBox(0, 0.0f, 0.0f, 4.0f, 2.0f, static_cast<float>(M_PI / 4.0), 0.9f),
            MakeBox(1, 0.2f, 0.1f, 4.0f, 2.0f, static_cast<float>(M_PI / 4.0), 0.8f),
        },
        0.1f, 10, {0});
 
    RunNmsCase(
        {
            MakeBox(0, 0.0f, 0.0f, 4.0f, 2.0f, static_cast<float>(M_PI / 4.0), 0.9f),
            MakeBox(1, 0.0f, 0.0f, 4.0f, 2.0f, static_cast<float>(-M_PI / 4.0), 0.8f),
        },
        0.1f, 10, {0});
 
    RunNmsCase(
        {
            MakeBox(0, 0.0f, 0.0f, 4.0f, 2.0f, static_cast<float>(M_PI / 4.0), 0.9f),
            MakeBox(1, 4.0f, 4.0f, 4.0f, 2.0f, static_cast<float>(M_PI / 4.0), 0.8f),
        },
        0.1f, 10, {0, 1});
 
    RunNmsCase(
        {
            MakeBox(0, 0.0f, 0.0f, 4.0f, 2.0f, 0.0f, 0.8f),
            MakeBox(1, 0.5f, 0.0f, 4.0f, 2.0f, 0.0f, 0.95f),
            MakeBox(2, 8.0f, 0.0f, 4.0f, 2.0f, 0.0f, 0.7f),
        },
        0.1f, 10, {1, 2});
 
    RunNmsCase(
        {
            MakeBox(0, 0.0f, 0.0f, 4.0f, 2.0f, 0.0f, 0.9f),
            MakeBox(1, 8.0f, 0.0f, 4.0f, 2.0f, 0.0f, 0.8f),
            MakeBox(2, 16.0f, 0.0f, 4.0f, 2.0f, 0.0f, 0.7f),
        },
        0.1f, 2, {0, 1});
  }
 
  {
    pcod_common::BoundingBox keep_box;
    keep_box.center = {0.0f, 0.0f};
    keep_box.length = 1.0f;
    keep_box.width = 1.0f;
    keep_box.existence_probability = 0.6f;
    keep_box.classification.push_back({0, 0.9f});
    keep_box.classification.push_back({1, 0.1f});
 
    pcod_common::BoundingBox drop_box;
    drop_box.center = {10.0f, 10.0f};  // ensure no IoU suppression interaction
    drop_box.length = 1.0f;
    drop_box.width = 1.0f;
    drop_box.existence_probability = 0.4f;
    drop_box.classification.push_back({0, 0.1f});
    drop_box.classification.push_back({1, 0.9f});
 
    std::vector<pcod_common::BoundingBox> boxes{keep_box, drop_box};
    pcod_common::NmsConfig config;
    config.score_thresholds = {0.1f, 0.9f};
    config.iou_threshold = 0.1f;
    config.max_detections = 10;
    config.internal_score_threshold = 0.5f;
 
    pcod_common::ApplyRotatedNms(boxes, config);
    assert(boxes.size() == 1);
    assert(std::abs(boxes[0].center[0]) < 1e-6f);
    assert(std::abs(boxes[0].center[1]) < 1e-6f);
  }
 
  return 0;
}

References pcod_common::ApplyRotatedNms(), pcod_common::BoundingBox::center, pcod_common::BoundingBox::classification, pcod_common::BoundingBox::existence_probability, pcod_common::NmsConfig::internal_score_threshold, pcod_common::NmsConfig::iou_threshold, pcod_common::BoundingBox::length, pcod_common::NmsConfig::max_detections, pcod_common::NmsConfig::score_thresholds, and pcod_common::BoundingBox::width.