From c37977394ec4d6574644e4c33dc7be390cf4e2c9 Mon Sep 17 00:00:00 2001
From: davidonyeali <98885753+davidonyeali@users.noreply.github.com>
Date: Tue, 4 Apr 2023 06:49:33 +0100
Subject: [PATCH] fix for sorting boxes in nonMaxSuppression

---
 .../org/pytorch/demo/objectdetection/PrePostProcessor.java  | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/ObjectDetection/app/src/main/java/org/pytorch/demo/objectdetection/PrePostProcessor.java b/ObjectDetection/app/src/main/java/org/pytorch/demo/objectdetection/PrePostProcessor.java
index b76bee14..3944b0cf 100644
--- a/ObjectDetection/app/src/main/java/org/pytorch/demo/objectdetection/PrePostProcessor.java
+++ b/ObjectDetection/app/src/main/java/org/pytorch/demo/objectdetection/PrePostProcessor.java
@@ -61,7 +61,11 @@ public int compare(Result o1, Result o2) {
                         return o1.score.compareTo(o2.score);
                     }
                 });
-
+        
+        // Collections.sort() sorts in ascending order
+        // reverse array to obtain boxes arranged in descending order
+        Collections.reverse(boxes);
+        
         ArrayList<Result> selected = new ArrayList<>();
         boolean[] active = new boolean[boxes.size()];
         Arrays.fill(active, true);