Added image classification implementation files and associated tests

2022-09-08 06:23:03 -07:00
parent 4dc4b19ddb
commit cb52432159
6 changed files with 764 additions and 1 deletions
@@ -47,3 +47,13 @@ py_library(
        "//mediapipe/tasks/python/core:optional_dependencies",
    ],
 )
+
+py_library(
+    name = "classifications",
+    srcs = ["classifications.py"],
+    deps = [
+        ":category",
+        "//mediapipe/tasks/cc/components/containers:classifications_py_pb2",
+        "//mediapipe/tasks/python/core:optional_dependencies",
+    ],
+)