Wolfram Function Repository
Instant-use add-on functions for the Wolfram Language
Function Repository Resource:
NonMaximum
Filter regions based on their overlap
Contributed by:
Giulio Alessandrini
ResourceFunction["NonMaximumSuppression"][{reg1,reg2,…}] keeps the rectangular regions regi with low overlap between each other. | |
ResourceFunction["NonMaximumSuppression"][{reg1,…}→{w1,…}] uses the weights wi to score the importance of each regi. | |
ResourceFunction["NonMaximumSuppression"][regions,property] gives the specified property for the filtered regions. |
Details and Options
ResourceFunction["NonMaximumSuppression"] is used to filter regions based on their mutual overlap and their weights.
Regions must be given in the form Rectangle[{xmin,ymin},{xmax,ymax}].
If the weights are not provided, regions are scored in linearly descending order with w1=1 and wn=1/n.
The following properties are supported:
| "Index" | position of the selected region |
| "Region" | selected region |
| "Weight" | the weight used to score the region |
| {prop1,…} | a list of properties |
| All | all the available properties |
The following options are supported:
| AcceptanceThreshold | Automatic | region acceptance threshold |
| MaxOverlapFraction | Automatic | maximum allowed overlap fraction |
| Method | "Hard" | method to use |
Possible values for Method are:
| "Hard" | remove regions with overlap greater than the specified threshold |
| "Soft" | iteratively rescore the regions based on their overlap |
Examples
Basic Examples (2)
Filter out rectangles from list that have a big overlap with other rectangles in the list:
| In[1]:= | ![ResourceFunction[
"NonMaximumSuppression"][{Rectangle[{0.1, 0.1}, {1, 1}], Rectangle[{0, 0}, {1, 1}], Rectangle[{0, 0}, {-2, -2}]}]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/158bac14697cf775.png){kind=small} |
| Out[1]= |  |
Define and display a list of rectangle shapes:
| In[2]:= | ![SeedRandom[123];
Short[(boxes = Flatten@Array[
Rectangle[2 {##} + RandomVariate[NormalDistribution[], 2], 2 {##} + RandomReal[{3, 9}, 2]] &, {5, 3}]), 2]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/5635aea5a302d02c.png) |
| Out[2]= |  |
| In[3]:= | ![g = Graphics[{Opacity[.05], EdgeForm[Black], boxes}, ImageSize -> Small]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/66cb110f7c6b6674.png){kind=small} |
| Out[3]= |  |
Delete rectangles with high overlap and display the results in red along with the original rectangles in black:
| In[4]:= | ![ResourceFunction["NonMaximumSuppression"][boxes];
Show[g, Graphics[{Opacity[0.3], Red, EdgeForm[Red], %}]]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/748fe31dba9ebfbc.png){kind=small} |
| Out[4]= |  |
Only keep non-overlapping rectangles:
| In[5]:= | ![ResourceFunction["NonMaximumSuppression"][boxes, MaxOverlapFraction -> 0];
Show[g, Graphics[{Opacity[0.3], Red, EdgeForm[Red], %}]]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/2c401ffcbaf5f39c.png){kind=small} |
| Out[5]= |  |
Scope (4)
Perform non-max suppression on a list of regions:
| In[6]:= | ![boxes = {
Rectangle[{3.241549223388991, 1.8258945791733319`}, {
10.659287859633295`, 10.773294493394275`}],
Rectangle[{0.4869914284730348, 4.321183643750274}, {5.369829888721986,
9.31386702637484}],
Rectangle[{2.2334201426158136`, 4.721638979849482}, {
5.291543408030995, 12.769601273219152`}],
Rectangle[{5.349846089859113, 2.8592299264608263`}, {
12.259520398084739`, 5.65464262378527}],
Rectangle[{5.419709227350725, 3.2032779499946393`}, {
8.019495864892848, 7.597470779871912}],
Rectangle[{2.9916342183590716`, 6.701712853508063}, {
12.82951197826295, 10.889573073474823`}],
Rectangle[{5.716124755621678, 4.016397691332317}, {
12.634485498683249`, 9.030579596384047}],
Rectangle[{4.318687518098674, 4.086673071150529}, {
12.377658857900416`, 10.295175285987018`}],
Rectangle[{5.254646844539921, 5.048321499467785}, {
11.505305608427992`, 13.227130047971752`}],
Rectangle[{8.231089286494171, 1.8404075028314466`}, {16.2229662869946,
8.789967125833199}],
Rectangle[{6.7403064092133205`, 4.533063177477215}, {
14.410877169797764`, 9.89745177054213}],
Rectangle[{9.862402933845159, 5.563175394985179}, {
11.276877537331307`, 11.97757563169206}],
Rectangle[{11.180782570286068`, 2.0918187606070013`}, {
15.93179551146674, 5.471116257515766}],
Rectangle[{9.160598972568872, 3.0721821586970166`}, {
16.105158304284622`, 11.003417304165653`}],
Rectangle[{9.199084080623214, 6.486880587531073}, {
18.967352233543956`, 13.733293159109362`}]};
Length[boxes]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/696760603da497d0.png){kind=small} |
| Out[6]= |  |
| In[7]:= | ![ResourceFunction["NonMaximumSuppression"][boxes] // Short](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/4fa87039a9e8b1e4.png){kind=small} |
| Out[7]= |  |
| In[8]:= | ![Graphics[{Opacity[0.05], EdgeForm[Black], boxes, Opacity[0.3], Red, EdgeForm[Red], %}, ImageSize -> Small]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/6fc967d4a9b797b7.png){kind=small} |
| Out[8]= |  |
Provide an explicit list of weights:
| In[9]:= | ![boxes = {
Rectangle[{3.241549223388991, 1.8258945791733319`}, {
10.659287859633295`, 10.773294493394275`}],
Rectangle[{0.4869914284730348, 4.321183643750274}, {5.369829888721986,
9.31386702637484}],
Rectangle[{2.2334201426158136`, 4.721638979849482}, {
5.291543408030995, 12.769601273219152`}],
Rectangle[{5.349846089859113, 2.8592299264608263`}, {
12.259520398084739`, 5.65464262378527}],
Rectangle[{5.419709227350725, 3.2032779499946393`}, {
8.019495864892848, 7.597470779871912}],
Rectangle[{2.9916342183590716`, 6.701712853508063}, {
12.82951197826295, 10.889573073474823`}],
Rectangle[{5.716124755621678, 4.016397691332317}, {
12.634485498683249`, 9.030579596384047}],
Rectangle[{4.318687518098674, 4.086673071150529}, {
12.377658857900416`, 10.295175285987018`}],
Rectangle[{5.254646844539921, 5.048321499467785}, {
11.505305608427992`, 13.227130047971752`}],
Rectangle[{8.231089286494171, 1.8404075028314466`}, {16.2229662869946,
8.789967125833199}],
Rectangle[{6.7403064092133205`, 4.533063177477215}, {
14.410877169797764`, 9.89745177054213}],
Rectangle[{9.862402933845159, 5.563175394985179}, {
11.276877537331307`, 11.97757563169206}],
Rectangle[{11.180782570286068`, 2.0918187606070013`}, {
15.93179551146674, 5.471116257515766}],
Rectangle[{9.160598972568872, 3.0721821586970166`}, {
16.105158304284622`, 11.003417304165653`}],
Rectangle[{9.199084080623214, 6.486880587531073}, {
18.967352233543956`, 13.733293159109362`}]};
scores = {0.18372941641066354`, 0.7792523485576668, 0.7552787064436872, 0.20397739818471128`, 0.29873901258951774`, 0.5385959520846411, 0.0019547594798015044`, 0.7158017739163653, 0.16959589408147346`, 0.8582668907417819, 0.21215626418505606`, 0.304448649385447, 0.33974292946852325`, 0.4268902847722864, 0.3182413575132059};
Length[boxes]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/6b8a822b44eb21f0.png) |
| Out[11]= |  |
Performing non-maximum suppression eliminated some of the rectangles:
| In[12]:= | ![result = ResourceFunction["NonMaximumSuppression"][boxes -> scores];
Length[result]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/48750a121a27861a.png){kind=small} |
| Out[13]= |  |
Show the result:
| In[14]:= | ![Graphics[{Opacity[0.05], EdgeForm[Black], boxes, Opacity[0.3], Red, EdgeForm[Red], result}, ImageSize -> Small]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/23feb0e627133942.png){kind=small} |
| Out[14]= |  |
Ask for a specific property:
| In[15]:= | ![boxes = {
Rectangle[{3.241549223388991, 1.8258945791733319`}, {
10.659287859633295`, 10.773294493394275`}],
Rectangle[{0.4869914284730348, 4.321183643750274}, {5.369829888721986,
9.31386702637484}],
Rectangle[{2.2334201426158136`, 4.721638979849482}, {
5.291543408030995, 12.769601273219152`}],
Rectangle[{5.349846089859113, 2.8592299264608263`}, {
12.259520398084739`, 5.65464262378527}],
Rectangle[{5.419709227350725, 3.2032779499946393`}, {
8.019495864892848, 7.597470779871912}],
Rectangle[{2.9916342183590716`, 6.701712853508063}, {
12.82951197826295, 10.889573073474823`}],
Rectangle[{5.716124755621678, 4.016397691332317}, {
12.634485498683249`, 9.030579596384047}],
Rectangle[{4.318687518098674, 4.086673071150529}, {
12.377658857900416`, 10.295175285987018`}],
Rectangle[{5.254646844539921, 5.048321499467785}, {
11.505305608427992`, 13.227130047971752`}],
Rectangle[{8.231089286494171, 1.8404075028314466`}, {16.2229662869946,
8.789967125833199}],
Rectangle[{6.7403064092133205`, 4.533063177477215}, {
14.410877169797764`, 9.89745177054213}],
Rectangle[{9.862402933845159, 5.563175394985179}, {
11.276877537331307`, 11.97757563169206}],
Rectangle[{11.180782570286068`, 2.0918187606070013`}, {
15.93179551146674, 5.471116257515766}],
Rectangle[{9.160598972568872, 3.0721821586970166`}, {
16.105158304284622`, 11.003417304165653`}],
Rectangle[{9.199084080623214, 6.486880587531073}, {
18.967352233543956`, 13.733293159109362`}]};
scores = {0.18372941641066354`, 0.7792523485576668, 0.7552787064436872, 0.20397739818471128`, 0.29873901258951774`, 0.5385959520846411, 0.0019547594798015044`, 0.7158017739163653, 0.16959589408147346`, 0.8582668907417819, 0.21215626418505606`, 0.304448649385447, 0.33974292946852325`, 0.4268902847722864, 0.3182413575132059};
ResourceFunction["NonMaximumSuppression"][boxes -> scores, "Index"]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/757f9118e7c7fafa.png) |
| Out[15]= |  |
The index is determined by the order of the scores for rectangles that contained in the result:
| In[16]:= | ![Reverse@Ordering[scores]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/3a545e0fd8f67d86.png){kind=small} |
| Out[16]= |  |
Ask for a list of properties:
| In[17]:= | ![boxes = {
Rectangle[{3.241549223388991, 1.8258945791733319`}, {
10.659287859633295`, 10.773294493394275`}],
Rectangle[{0.4869914284730348, 4.321183643750274}, {5.369829888721986,
9.31386702637484}],
Rectangle[{2.2334201426158136`, 4.721638979849482}, {
5.291543408030995, 12.769601273219152`}],
Rectangle[{5.349846089859113, 2.8592299264608263`}, {
12.259520398084739`, 5.65464262378527}],
Rectangle[{5.419709227350725, 3.2032779499946393`}, {
8.019495864892848, 7.597470779871912}],
Rectangle[{2.9916342183590716`, 6.701712853508063}, {
12.82951197826295, 10.889573073474823`}],
Rectangle[{5.716124755621678, 4.016397691332317}, {
12.634485498683249`, 9.030579596384047}],
Rectangle[{4.318687518098674, 4.086673071150529}, {
12.377658857900416`, 10.295175285987018`}],
Rectangle[{5.254646844539921, 5.048321499467785}, {
11.505305608427992`, 13.227130047971752`}],
Rectangle[{8.231089286494171, 1.8404075028314466`}, {16.2229662869946,
8.789967125833199}],
Rectangle[{6.7403064092133205`, 4.533063177477215}, {
14.410877169797764`, 9.89745177054213}],
Rectangle[{9.862402933845159, 5.563175394985179}, {
11.276877537331307`, 11.97757563169206}],
Rectangle[{11.180782570286068`, 2.0918187606070013`}, {
15.93179551146674, 5.471116257515766}],
Rectangle[{9.160598972568872, 3.0721821586970166`}, {
16.105158304284622`, 11.003417304165653`}],
Rectangle[{9.199084080623214, 6.486880587531073}, {
18.967352233543956`, 13.733293159109362`}]};
scores = {0.18372941641066354`, 0.7792523485576668, 0.7552787064436872, 0.20397739818471128`, 0.29873901258951774`, 0.5385959520846411, 0.0019547594798015044`, 0.7158017739163653, 0.16959589408147346`, 0.8582668907417819, 0.21215626418505606`, 0.304448649385447, 0.33974292946852325`, 0.4268902847722864, 0.3182413575132059};
ResourceFunction["NonMaximumSuppression"][
boxes -> scores, {"Index", "Region"}]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/72c0fdd1fd5952db.png) |
| Out[19]= |  |
Ask for all properties:
| In[20]:= | ![ResourceFunction["NonMaximumSuppression"][boxes -> scores, All]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/414a8043ba10c936.png){kind=small} |
| Out[20]= |  |
Options (6)
AcceptanceThreshold (2)
Specify an AcceptanceThreshold for the regions:
| In[21]:= | ![boxes = {
Rectangle[{3.241549223388991, 1.8258945791733319`}, {
10.659287859633295`, 10.773294493394275`}],
Rectangle[{0.4869914284730348, 4.321183643750274}, {5.369829888721986,
9.31386702637484}],
Rectangle[{2.2334201426158136`, 4.721638979849482}, {
5.291543408030995, 12.769601273219152`}],
Rectangle[{5.349846089859113, 2.8592299264608263`}, {
12.259520398084739`, 5.65464262378527}],
Rectangle[{5.419709227350725, 3.2032779499946393`}, {
8.019495864892848, 7.597470779871912}],
Rectangle[{2.9916342183590716`, 6.701712853508063}, {
12.82951197826295, 10.889573073474823`}],
Rectangle[{5.716124755621678, 4.016397691332317}, {
12.634485498683249`, 9.030579596384047}],
Rectangle[{4.318687518098674, 4.086673071150529}, {
12.377658857900416`, 10.295175285987018`}],
Rectangle[{5.254646844539921, 5.048321499467785}, {
11.505305608427992`, 13.227130047971752`}],
Rectangle[{8.231089286494171, 1.8404075028314466`}, {16.2229662869946,
8.789967125833199}],
Rectangle[{6.7403064092133205`, 4.533063177477215}, {
14.410877169797764`, 9.89745177054213}],
Rectangle[{9.862402933845159, 5.563175394985179}, {
11.276877537331307`, 11.97757563169206}],
Rectangle[{11.180782570286068`, 2.0918187606070013`}, {
15.93179551146674, 5.471116257515766}],
Rectangle[{9.160598972568872, 3.0721821586970166`}, {
16.105158304284622`, 11.003417304165653`}],
Rectangle[{9.199084080623214, 6.486880587531073}, {
18.967352233543956`, 13.733293159109362`}]};
scores = {0.18372941641066354`, 0.7792523485576668, 0.7552787064436872, 0.20397739818471128`, 0.29873901258951774`, 0.5385959520846411, 0.0019547594798015044`, 0.7158017739163653, 0.16959589408147346`, 0.8582668907417819, 0.21215626418505606`, 0.304448649385447, 0.33974292946852325`, 0.4268902847722864, 0.3182413575132059};
ResourceFunction["NonMaximumSuppression"][boxes -> scores, AcceptanceThreshold -> .5]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/68903062dbab1f8f.png) |
| Out[21]= |  |
| In[22]:= | ![Graphics[{Opacity[0.05], EdgeForm[Black], boxes, Opacity[0.3], Red, EdgeForm[Red], %}, ImageSize -> Small]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/2cee53a678bf9632.png){kind=small} |
| Out[22]= |  |
Specify values to use before and after the overlap filtering:
| In[23]:= | ![boxes = {
Rectangle[{3.241549223388991, 1.8258945791733319`}, {
10.659287859633295`, 10.773294493394275`}],
Rectangle[{0.4869914284730348, 4.321183643750274}, {5.369829888721986,
9.31386702637484}],
Rectangle[{2.2334201426158136`, 4.721638979849482}, {
5.291543408030995, 12.769601273219152`}],
Rectangle[{5.349846089859113, 2.8592299264608263`}, {
12.259520398084739`, 5.65464262378527}],
Rectangle[{5.419709227350725, 3.2032779499946393`}, {
8.019495864892848, 7.597470779871912}],
Rectangle[{2.9916342183590716`, 6.701712853508063}, {
12.82951197826295, 10.889573073474823`}],
Rectangle[{5.716124755621678, 4.016397691332317}, {
12.634485498683249`, 9.030579596384047}],
Rectangle[{4.318687518098674, 4.086673071150529}, {
12.377658857900416`, 10.295175285987018`}],
Rectangle[{5.254646844539921, 5.048321499467785}, {
11.505305608427992`, 13.227130047971752`}],
Rectangle[{8.231089286494171, 1.8404075028314466`}, {16.2229662869946,
8.789967125833199}],
Rectangle[{6.7403064092133205`, 4.533063177477215}, {
14.410877169797764`, 9.89745177054213}],
Rectangle[{9.862402933845159, 5.563175394985179}, {
11.276877537331307`, 11.97757563169206}],
Rectangle[{11.180782570286068`, 2.0918187606070013`}, {
15.93179551146674, 5.471116257515766}],
Rectangle[{9.160598972568872, 3.0721821586970166`}, {
16.105158304284622`, 11.003417304165653`}],
Rectangle[{9.199084080623214, 6.486880587531073}, {
18.967352233543956`, 13.733293159109362`}]};
scores = {0.18372941641066354`, 0.7792523485576668, 0.7552787064436872, 0.20397739818471128`, 0.29873901258951774`, 0.5385959520846411, 0.0019547594798015044`, 0.7158017739163653, 0.16959589408147346`, 0.8582668907417819, 0.21215626418505606`, 0.304448649385447, 0.33974292946852325`, 0.4268902847722864, 0.3182413575132059};
ResourceFunction["NonMaximumSuppression"][boxes -> scores, Method -> "Soft", AcceptanceThreshold -> {.2, .5}]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/248380350216c74d.png) |
| Out[23]= |  |
| In[24]:= | ![Graphics[{Opacity[0.05], EdgeForm[Black], boxes, Opacity[0.3], Red, EdgeForm[Red], %}, ImageSize -> Small]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/20516e15e5ad07d6.png){kind=small} |
| Out[24]= |  |
MaxOverlapFraction (2)
Use MaxOverlapFraction→f to only keep regions with an overlap smaller than f:
| In[25]:= | ![boxes = {
Rectangle[{3.241549223388991, 1.8258945791733319`}, {
10.659287859633295`, 10.773294493394275`}],
Rectangle[{0.4869914284730348, 4.321183643750274}, {5.369829888721986,
9.31386702637484}],
Rectangle[{2.2334201426158136`, 4.721638979849482}, {
5.291543408030995, 12.769601273219152`}],
Rectangle[{5.349846089859113, 2.8592299264608263`}, {
12.259520398084739`, 5.65464262378527}],
Rectangle[{5.419709227350725, 3.2032779499946393`}, {
8.019495864892848, 7.597470779871912}],
Rectangle[{2.9916342183590716`, 6.701712853508063}, {
12.82951197826295, 10.889573073474823`}],
Rectangle[{5.716124755621678, 4.016397691332317}, {
12.634485498683249`, 9.030579596384047}],
Rectangle[{4.318687518098674, 4.086673071150529}, {
12.377658857900416`, 10.295175285987018`}],
Rectangle[{5.254646844539921, 5.048321499467785}, {
11.505305608427992`, 13.227130047971752`}],
Rectangle[{8.231089286494171, 1.8404075028314466`}, {16.2229662869946,
8.789967125833199}],
Rectangle[{6.7403064092133205`, 4.533063177477215}, {
14.410877169797764`, 9.89745177054213}],
Rectangle[{9.862402933845159, 5.563175394985179}, {
11.276877537331307`, 11.97757563169206}],
Rectangle[{11.180782570286068`, 2.0918187606070013`}, {
15.93179551146674, 5.471116257515766}],
Rectangle[{9.160598972568872, 3.0721821586970166`}, {
16.105158304284622`, 11.003417304165653`}],
Rectangle[{9.199084080623214, 6.486880587531073}, {
18.967352233543956`, 13.733293159109362`}]};
scores = {0.18372941641066354`, 0.7792523485576668, 0.7552787064436872, 0.20397739818471128`, 0.29873901258951774`, 0.5385959520846411, 0.0019547594798015044`, 0.7158017739163653, 0.16959589408147346`, 0.8582668907417819, 0.21215626418505606`, 0.304448649385447, 0.33974292946852325`, 0.4268902847722864, 0.3182413575132059};
ResourceFunction["NonMaximumSuppression"][boxes -> scores] // Length](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/2e1a65665dc80391.png) |
| Out[25]= |  |
| In[26]:= | ![ResourceFunction["NonMaximumSuppression"][boxes -> scores, MaxOverlapFraction -> .2] // Length](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/381f60610749595d.png){kind=small} |
| Out[26]= |  |
If soft NMS is used, MaxOverlapFraction is used to defined the scaling behaviour:
| In[27]:= | ![boxes = {
Rectangle[{3.241549223388991, 1.8258945791733319`}, {
10.659287859633295`, 10.773294493394275`}],
Rectangle[{0.4869914284730348, 4.321183643750274}, {5.369829888721986,
9.31386702637484}],
Rectangle[{2.2334201426158136`, 4.721638979849482}, {
5.291543408030995, 12.769601273219152`}],
Rectangle[{5.349846089859113, 2.8592299264608263`}, {
12.259520398084739`, 5.65464262378527}],
Rectangle[{5.419709227350725, 3.2032779499946393`}, {
8.019495864892848, 7.597470779871912}],
Rectangle[{2.9916342183590716`, 6.701712853508063}, {
12.82951197826295, 10.889573073474823`}],
Rectangle[{5.716124755621678, 4.016397691332317}, {
12.634485498683249`, 9.030579596384047}],
Rectangle[{4.318687518098674, 4.086673071150529}, {
12.377658857900416`, 10.295175285987018`}],
Rectangle[{5.254646844539921, 5.048321499467785}, {
11.505305608427992`, 13.227130047971752`}],
Rectangle[{8.231089286494171, 1.8404075028314466`}, {16.2229662869946,
8.789967125833199}],
Rectangle[{6.7403064092133205`, 4.533063177477215}, {
14.410877169797764`, 9.89745177054213}],
Rectangle[{9.862402933845159, 5.563175394985179}, {
11.276877537331307`, 11.97757563169206}],
Rectangle[{11.180782570286068`, 2.0918187606070013`}, {
15.93179551146674, 5.471116257515766}],
Rectangle[{9.160598972568872, 3.0721821586970166`}, {
16.105158304284622`, 11.003417304165653`}],
Rectangle[{9.199084080623214, 6.486880587531073}, {
18.967352233543956`, 13.733293159109362`}]};
scores = {0.18372941641066354`, 0.7792523485576668, 0.7552787064436872, 0.20397739818471128`, 0.29873901258951774`, 0.5385959520846411, 0.0019547594798015044`, 0.7158017739163653, 0.16959589408147346`, 0.8582668907417819, 0.21215626418505606`, 0.304448649385447, 0.33974292946852325`, 0.4268902847722864, 0.3182413575132059};
ResourceFunction["NonMaximumSuppression"][boxes -> scores, MaxOverlapFraction -> 0, Method -> "Soft"] // Length](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/178d5d072e7f040c.png) |
| Out[27]= |  |
If the a scaling value is given explicitly, the option is ignored:
| In[28]:= | ![ResourceFunction["NonMaximumSuppression"][boxes -> scores, MaxOverlapFraction -> 0, Method -> {"Soft", 1}] // Length](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/32306129370f9fcc.png){kind=small} |
| Out[28]= |  |
Method (2)
By default, regions are removed using a hard overlap threshold:
| In[29]:= | ![boxes = {
Rectangle[{3.241549223388991, 1.8258945791733319`}, {
10.659287859633295`, 10.773294493394275`}],
Rectangle[{0.4869914284730348, 4.321183643750274}, {5.369829888721986,
9.31386702637484}],
Rectangle[{2.2334201426158136`, 4.721638979849482}, {
5.291543408030995, 12.769601273219152`}],
Rectangle[{5.349846089859113, 2.8592299264608263`}, {
12.259520398084739`, 5.65464262378527}],
Rectangle[{5.419709227350725, 3.2032779499946393`}, {
8.019495864892848, 7.597470779871912}],
Rectangle[{2.9916342183590716`, 6.701712853508063}, {
12.82951197826295, 10.889573073474823`}],
Rectangle[{5.716124755621678, 4.016397691332317}, {
12.634485498683249`, 9.030579596384047}],
Rectangle[{4.318687518098674, 4.086673071150529}, {
12.377658857900416`, 10.295175285987018`}],
Rectangle[{5.254646844539921, 5.048321499467785}, {
11.505305608427992`, 13.227130047971752`}],
Rectangle[{8.231089286494171, 1.8404075028314466`}, {16.2229662869946,
8.789967125833199}],
Rectangle[{6.7403064092133205`, 4.533063177477215}, {
14.410877169797764`, 9.89745177054213}],
Rectangle[{9.862402933845159, 5.563175394985179}, {
11.276877537331307`, 11.97757563169206}],
Rectangle[{11.180782570286068`, 2.0918187606070013`}, {
15.93179551146674, 5.471116257515766}],
Rectangle[{9.160598972568872, 3.0721821586970166`}, {
16.105158304284622`, 11.003417304165653`}],
Rectangle[{9.199084080623214, 6.486880587531073}, {
18.967352233543956`, 13.733293159109362`}]};
scores = {0.18372941641066354`, 0.7792523485576668, 0.7552787064436872, 0.20397739818471128`, 0.29873901258951774`, 0.5385959520846411, 0.0019547594798015044`, 0.7158017739163653, 0.16959589408147346`, 0.8582668907417819, 0.21215626418505606`, 0.304448649385447, 0.33974292946852325`, 0.4268902847722864, 0.3182413575132059};
result = ResourceFunction["NonMaximumSuppression"][boxes -> scores]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/65b45c8b00702457.png) |
| Out[31]= |  |
| In[32]:= | ![Graphics[{Opacity[0.05], EdgeForm[Black], boxes, Opacity[0.3], Red, EdgeForm[Red], result}, ImageSize -> Small]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/789ddc4057e6f5bb.png){kind=small} |
| Out[32]= |  |
Filter boxes using a soft NMS that scales the weights based on the overlap:
| In[33]:= | ![ResourceFunction["NonMaximumSuppression"][boxes -> scores, Method -> "Soft"];
Graphics[{Opacity[0.05], EdgeForm[Black], boxes, Opacity[0.3], Red, EdgeForm[Red], %}, ImageSize -> Small]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/57ec7e4205151498.png) |
| Out[33]= |  |
Specify the scaling value used to update the regions weights:
| In[34]:= | ![ResourceFunction["NonMaximumSuppression"][boxes -> scores, Method -> {"Soft", .05}];
Graphics[{Opacity[0.05], EdgeForm[Black], boxes, Opacity[0.3], Red, EdgeForm[Red], %}, ImageSize -> Small]](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/002559fdd9b5b9b5.png) |
| Out[34]= |  |
Soft NMS can become very slow if boxes are not removed:
| In[35]:= | ![test = Rectangle[First[#], Total[#] + 1] & /@ RandomReal[10, {5000, 2, 2}];](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/3cc33eb102341b4f.png){kind=small} |
| In[36]:= | ![ResourceFunction["NonMaximumSuppression"][test, AcceptanceThreshold -> 0, Method -> "Hard"] // RepeatedTiming // First](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/73a5d99c1ffe10e9.png){kind=small} |
| Out[36]= |  |
| In[37]:= | ![ResourceFunction["NonMaximumSuppression"][test, AcceptanceThreshold -> 0, Method -> "Soft"] // RepeatedTiming // First](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/7e4d7613c0f97cb1.png){kind=small} |
| Out[37]= |  |
Even a small threshold can significantly speed-up the computation:
| In[38]:= | ![ResourceFunction["NonMaximumSuppression"][test, AcceptanceThreshold -> .1, Method -> "Soft"] // RepeatedTiming // First](https://www.wolframcloud.com/obj/resourcesystem/images/1a4/1a4d3d13-f50c-4ae2-9c5f-860c451341c5/0dc17753be0a0e82.png){kind=small} |
| Out[38]= |  |
Version History
- 1.0.0 – 14 December 2021
Source Metadata
Related Symbols
License Information
This work is licensed under a Creative Commons Attribution 4.0 International License