From 78c7ae31b291a0af0f17cae61b7cde514ab20cf4 Mon Sep 17 00:00:00 2001 From: Smirnov Egor Date: Tue, 5 Oct 2021 16:45:09 +0300 Subject: [PATCH] add test data for SoftNMS --- testdata/dnn/soft_nms_reference.yml | 31 +++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 testdata/dnn/soft_nms_reference.yml diff --git a/testdata/dnn/soft_nms_reference.yml b/testdata/dnn/soft_nms_reference.yml new file mode 100644 index 000000000..19c3aff4f --- /dev/null +++ b/testdata/dnn/soft_nms_reference.yml @@ -0,0 +1,31 @@ +%YAML:1.0 +--- +boxes: [ 409, 55, 82, 25, 157, 71, 65, 25, 427, 25, 35, 23, 131, 133, + 322, 94, 403, 55, 80, 24, 155, 72, 66, 22, 416, 28, 48, 24, 50, 106, + 485, 155, 78, 286, 460, 64, 176, 76, 64, 17, 0, 0, 598, 346, 36, 78, + 58, 46, 24, 6, 258, 164, 98, 189, 423, 145, 407, 56, 81, 25, 159, 69, + 62, 25, 431, 28, 36, 21, 123, 121, 336, 113, 41, 51, 181, 72, 430, + 27, 36, 21, 399, 49, 86, 30, 155, 72, 61, 24, 84, 102, 426, 166, 128, + 0, 336, 347, 127, 251, 337, 101, 390, 22, 78, 39, 432, 26, 38, 23, + 156, 72, 61, 23, 414, 55, 79, 22, 215, 135, 165, 88, 276, 166, 84, + 48, 301, 22, 81, 45, 402, 59, 77, 28, 521, 11, 76, 35, 12, 73, 48, + 54, 193, 163, 197, 36, 146, 127, 127, 49, 44, 75, 60, 49 ] +probs: [ 1., 1., 9.50373888e-01, 2.62993928e-02, 9.99991059e-01, + 9.97794747e-01, 1.65874109e-01, 5.08980490e-02, 2.25577131e-02, + 2.18296256e-02, 1.93440001e-02, 1.31233213e-02, 1.28434300e-02, + 1.08395228e-02, 1., 9.99996901e-01, 9.98169780e-01, 3.46690901e-02, + 1.00121163e-02, 9.99895453e-01, 9.68177438e-01, 9.65382040e-01, + 6.64785877e-02, 2.35476252e-02, 1.25701502e-02, 1.25500374e-02, 1., + 1., 9.99995470e-01, 2.12208077e-01, 3.39371487e-02, 2.36911718e-02, + 1.37866093e-02, 1.24980714e-02, 1.21721951e-02, 1.16886804e-02, + 1.12007223e-02, 1.11661423e-02 ] +indices: [ 0, 1, 26, 20, 2, 15, 28, 16, 21, 29, 14, 19, 5, 4, 22, 27, 6, + 30, 7, 31, 8, 17, 23, 11, 33, 12, 34, 36 ] +updated_scores: [ 1.000000000000000, 1.000000000000000, 1.000000000000000, + 0.634520173072815, 0.624747872352600, 0.514399647712708, 0.376939475536346, + 0.329439252614975, 0.283865988254547, 0.212208077311516, 0.190269082784653, + 0.146278440952301, 0.118331797420979, 0.095820419490337, 0.063734129071236, + 0.053496349602938, 0.050462152808905, 0.031316649168730, 0.024680756032467, + 0.023691171780229, 0.022557713091373, 0.017200922593474, 0.014761860482395, + 0.013121815398335, 0.012498071417212, 0.011637533083558, 0.011319587938488, + 0.010369089432061 ]