fix: make ios labels match the ts enum

software-mansion · Dec 16, 2024 · 0d631a6 · 0d631a6
1 parent 35661b7
commit 0d631a6
Showing 1 changed file with 31 additions and 31 deletions.
diff --git a/ios/RnExecutorch/utils/llms/Constants.mm b/ios/RnExecutorch/utils/llms/Constants.mm
@@ -20,34 +20,34 @@
                                                                  encoding:NSUTF8StringEncoding];
 
 const std::unordered_map<int, std::string> cocoLabelsMap = {
-    {1, "person"},         {2, "bicycle"},       {3, "car"},
-    {4, "motorcycle"},     {5, "airplane"},      {6, "bus"},
-    {7, "train"},          {8, "truck"},         {9, "boat"},
-    {10, "traffic light"}, {11, "fire hydrant"}, {12, "street sign"},
-    {13, "stop sign"},     {14, "parking"},      {15, "bench"},
-    {16, "bird"},          {17, "cat"},          {18, "dog"},
-    {19, "horse"},         {20, "sheep"},        {21, "cow"},
-    {22, "elephant"},      {23, "bear"},         {24, "zebra"},
-    {25, "giraffe"},       {26, "hat"},          {27, "backpack"},
-    {28, "umbrella"},      {29, "shoe"},         {30, "eye"},
-    {31, "handbag"},       {32, "tie"},          {33, "suitcase"},
-    {34, "frisbee"},       {35, "skis"},         {36, "snowboard"},
-    {37, "sports"},        {38, "kite"},         {39, "baseball"},
-    {40, "baseball"},      {41, "skateboard"},   {42, "surfboard"},
-    {43, "tennis racket"}, {44, "bottle"},       {45, "plate"},
-    {46, "wine glass"},    {47, "cup"},          {48, "fork"},
-    {49, "knife"},         {50, "spoon"},        {51, "bowl"},
-    {52, "banana"},        {53, "apple"},        {54, "sandwich"},
-    {55, "orange"},        {56, "broccoli"},     {57, "carrot"},
-    {58, "hot dog"},       {59, "pizza"},        {60, "donut"},
-    {61, "cake"},          {62, "chair"},        {63, "couch"},
-    {64, "potted plant"},  {65, "bed"},          {66, "mirror"},
-    {67, "dining table"},  {68, "window"},       {69, "desk"},
-    {70, "toilet"},        {71, "door"},         {72, "tv"},
-    {73, "laptop"},        {74, "mouse"},        {75, "remote"},
-    {76, "keyboard"},      {77, "cell phone"},   {78, "microwave"},
-    {79, "oven"},          {80, "toaster"},      {81, "sink"},
-    {82, "refrigerator"},  {83, "blender"},      {84, "book"},
-    {85, "clock"},         {86, "vase"},         {87, "scissors"},
-    {88, "teddy bear"},    {89, "hair drier"},   {90, "toothbrush"},
-    {91, "hair brush"}};
+    {1, "PERSON"},         {2, "BICYCLE"},       {3, "CAR"},
+    {4, "MOTORCYCLE"},     {5, "AIRPLANE"},      {6, "BUS"},
+    {7, "TRAIN"},          {8, "TRUCK"},         {9, "BOAT"},
+    {10, "TRAFFIC_LIGHT"}, {11, "FIRE_HYDRANT"}, {12, "STREET_SIGN"},
+    {13, "STOP_SIGN"},     {14, "PARKING"},      {15, "BENCH"},
+    {16, "BIRD"},          {17, "CAT"},          {18, "DOG"},
+    {19, "HORSE"},         {20, "SHEEP"},        {21, "COW"},
+    {22, "ELEPHANT"},      {23, "BEAR"},         {24, "ZEBRA"},
+    {25, "GIRAFFE"},       {26, "HAT"},          {27, "BACKPACK"},
+    {28, "UMBRELLA"},      {29, "SHOE"},         {30, "EYE"},
+    {31, "HANDBAG"},       {32, "TIE"},          {33, "SUITCASE"},
+    {34, "FRISBEE"},       {35, "SKIS"},         {36, "SNOWBOARD"},
+    {37, "SPORTS"},        {38, "KITE"},         {39, "BASEBALL"},
+    {40, "BASEBALL"},      {41, "SKATEBOARD"},   {42, "SURFBOARD"},
+    {43, "TENNIS_RACKET"}, {44, "BOTTLE"},       {45, "PLATE"},
+    {46, "WINE_GLASS"},    {47, "CUP"},          {48, "FORK"},
+    {49, "KNIFE"},         {50, "SPOON"},        {51, "BOWL"},
+    {52, "BANANA"},        {53, "APPLE"},        {54, "SANDWICH"},
+    {55, "ORANGE"},        {56, "BROCCOLI"},     {57, "CARROT"},
+    {58, "HOT_DOG"},       {59, "PIZZA"},        {60, "DONUT"},
+    {61, "CAKE"},          {62, "CHAIR"},        {63, "COUCH"},
+    {64, "POTTED_PLANT"},  {65, "BED"},          {66, "MIRROR"},
+    {67, "DINING_TABLE"},  {68, "WINDOW"},       {69, "DESK"},
+    {70, "TOILET"},        {71, "DOOR"},         {72, "TV"},
+    {73, "LAPTOP"},        {74, "MOUSE"},        {75, "REMOTE"},
+    {76, "KEYBOARD"},      {77, "CELL_PHONE"},   {78, "MICROWAVE"},
+    {79, "OVEN"},          {80, "TOASTER"},      {81, "SINK"},
+    {82, "REFRIGERATOR"},  {83, "BLENDER"},      {84, "BOOK"},
+    {85, "CLOCK"},         {86, "VASE"},         {87, "SCISSORS"},
+    {88, "TEDDY_BEAR"},    {89, "HAIR_DRIER"},   {90, "TOOTHBRUSH"},
+    {91, "HAIR_BRUSH"}};