Add convertFromClassifications() helper

PiperOrigin-RevId: 519181016

mediapipe/tasks/web/components/processors/BUILD

@@ -34,6 +34,7 @@ mediapipe_ts_library(
     name = "classifier_result",
     srcs = ["classifier_result.ts"],
     deps = [
+        "//mediapipe/framework/formats:classification_jspb_proto",
         "//mediapipe/tasks/cc/components/containers/proto:classifications_jspb_proto",
         "//mediapipe/tasks/web/components/containers:classification_result",
     ],

mediapipe/tasks/web/components/processors/classifier_result.ts

@@ -14,6 +14,7 @@
  * limitations under the License.
  */
 
+import {Classification as ClassificationProto} from '../../../../framework/formats/classification_pb';
 import {ClassificationResult as ClassificationResultProto, Classifications as ClassificationsProto} from '../../../../tasks/cc/components/containers/proto/classifications_pb';
 import {ClassificationResult, Classifications} from '../../../../tasks/web/components/containers/classification_result';
 
@@ -21,28 +22,36 @@ const DEFAULT_INDEX = -1;
 const DEFAULT_SCORE = 0.0;
 
 /**
- * Converts a Classifications proto to a Classifications object.
+ * Converts a list of Classification protos to a Classifications object.
  */
-function convertFromClassificationsProto(source: ClassificationsProto):
-    Classifications {
-  const categories =
-      source.getClassificationList()?.getClassificationList().map(
-          classification => {
+export function convertFromClassifications(
+    classifications: ClassificationProto[], headIndex = DEFAULT_INDEX,
+    headName = ''): Classifications {
+  const categories = classifications.map(classification => {
     return {
       index: classification.getIndex() ?? DEFAULT_INDEX,
       score: classification.getScore() ?? DEFAULT_SCORE,
       categoryName: classification.getLabel() ?? '',
       displayName: classification.getDisplayName() ?? '',
     };
-          }) ??
-      [];
+  });
   return {
     categories,
-    headIndex: source.getHeadIndex() ?? DEFAULT_INDEX,
-    headName: source.getHeadName() ?? '',
+    headIndex,
+    headName,
   };
 }
 
+/**
+ * Converts a Classifications proto to a Classifications object.
+ */
+function convertFromClassificationsProto(source: ClassificationsProto):
+    Classifications {
+  return convertFromClassifications(
+      source.getClassificationList()?.getClassificationList() ?? [],
+      source.getHeadIndex(), source.getHeadName());
+}
+
 /**
  * Converts a ClassificationResult proto to a ClassificationResult object.
  */