Change segmentation labels to uppercase on Android

Author: JakubGonera

android/src/main/java/com/swmansion/rnexecutorch/models/imageSegmentation/Constants.kt

@@ -1,8 +1,26 @@
 package com.swmansion.rnexecutorch.models.imagesegmentation
 
-val deeplabv3_resnet50_labels: Array<String> = arrayOf(
-    "background", "aeroplane", "bicycle", "bird", "boat",
-    "bottle", "bus", "car", "cat", "chair", "cow", "diningtable",
-    "dog", "horse", "motorbike", "person", "pottedplant", "sheep",
-    "sofa", "train", "tvmonitor"
-)
+val deeplabv3_resnet50_labels: Array<String> =
+  arrayOf(
+    "BACKGROUND",
+    "AEROPLANE",
+    "BICYCLE",
+    "BIRD",
+    "BOAT",
+    "BOTTLE",
+    "BUS",
+    "CAR",
+    "CAT",
+    "CHAIR",
+    "COW",
+    "DININGTABLE",
+    "DOG",
+    "HORSE",
+    "MOTORBIKE",
+    "PERSON",
+    "POTTEDPLANT",
+    "SHEEP",
+    "SOFA",
+    "TRAIN",
+    "TVMONITOR",
+  )

android/src/main/java/com/swmansion/rnexecutorch/models/imageSegmentation/ImageSegmentationModel.kt

@@ -94,15 +94,15 @@ class ImageSegmentationModel(
     classesOfInterest: ReadableArray,
     resize: Boolean,
   ): WritableMap {
-    val output = output[0].toTensor().dataAsFloatArray
+    val outputData = output[0].toTensor().dataAsFloatArray
     val modelSize = getModelImageSize()
     val numLabels = deeplabv3_resnet50_labels.size
 
-    require(output.count() == (numLabels * modelSize.height * modelSize.width).toInt()) { "Model generated unexpected output size." }
+    require(outputData.count() == (numLabels * modelSize.height * modelSize.width).toInt()) { "Model generated unexpected output size." }
 
     val outputSize = if (resize) originalSize else modelSize
 
-    val extractedResults = extractResults(output, numLabels, resize)
+    val extractedResults = extractResults(outputData, numLabels, resize)
 
     val argMax = adjustScoresPerPixel(extractedResults, numLabels, outputSize)
 
@@ -124,7 +124,7 @@ class ImageSegmentationModel(
     }
 
     res.putArray(
-      "argmax",
+      "ARGMAX",
       ArrayUtils.createReadableArrayFromIntArray(argMax),
     )