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 selected = new ArrayList<>(); boolean[] active = new boolean[boxes.size()]; Arrays.fill(active, true);