(7/n) Release DeepFusion (https://arxiv.org/pdf/2203.08195.pdf)

PiperOrigin-RevId: 455700495
tensorflow · Jun 17, 2022 · 6b75b05 · 6b75b05
1 parent b06f47c
commit 6b75b05
Show file tree

Hide file tree

Showing 3 changed files with 478 additions and 0 deletions.
diff --git a/lingvo/tasks/car/BUILD b/lingvo/tasks/car/BUILD
@@ -382,6 +382,7 @@ py_library(
     srcs_version = "PY3",
     deps = [
         ":builder_lib",
+        ":car_layers",
         ":detection_3d_lib",
         ":geometry",
         ":point_detector",
@@ -597,3 +598,17 @@ py_library(
         # Implicit tensorflow_probability dependency.
     ],
 )
+
+py_library(
+    name = "deep_fusion",
+    srcs = ["deep_fusion.py"],
+    srcs_version = "PY3",
+    deps = [
+        ":car_lib",
+        ":pillars",
+        "//lingvo:compat",
+        "//lingvo/core:base_layer",
+        "//lingvo/core:py_utils",
+        # Implicit numpy dependency.
+    ],
+)