Skip to content

Commit 7c503b8

Browse files
Waymo Researchpeisun1115
authored andcommitted
Merged commit includes the following changes:
348491037 by Waymo Research: Fix tracking metrics build fail on github. -- 348482312 by Waymo Research: Doc fix -- PiperOrigin-RevId: 348491037
1 parent 945d43c commit 7c503b8

File tree

2 files changed

+47
-7
lines changed

2 files changed

+47
-7
lines changed

docs/quick_start.md

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ As a first step, please visit https://www.waymo.com/open to gain access to the W
1212
## System Requirements
1313

1414
* g++ 5 or higher.
15-
* TensorFlow 1.15.0, 2.0.0, 2.1.0
15+
* TensorFlow 1.15.0, 2.0.0, 2.1.0, 2.2.0, 2.3.0
1616

1717
The code has two main parts. One is a utility written in C++ to compute the
1818
evaluation metrics. The other part is a set of
@@ -30,12 +30,11 @@ install it in most cases. Please see these
3030
[instructions](https://docs.bazel.build/versions/master/install.html) for other
3131
ways to install Bazel. We assume you have Python installed.
3232
```bash
33-
sudo apt-get
34-
install --assume-yes pkg-config zip g++ zlib1g-dev unzip python3 python3-pip
35-
BAZEL_VERSION=3.1.0 wget
36-
https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-installer-linux-x86_64.sh
37-
sudo bash bazel-${BAZEL_VERSION}-installer-linux-x86_64.sh sudo apt install
38-
build-essential
33+
sudo apt-get install --assume-yes pkg-config zip g++ zlib1g-dev unzip python3 python3-pip
34+
BAZEL_VERSION=3.1.0
35+
wget https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-installer-linux-x86_64.sh
36+
sudo bash bazel-${BAZEL_VERSION}-installer-linux-x86_64.sh
37+
sudo apt install build-essential
3938
```
4039

4140
Configure .bazelrc. `./configure.sh`

waymo_open_dataset/metrics/ops/BUILD

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ py_library(
2828
srcs = ["py_metrics_ops.py"],
2929
data = [
3030
":detection_metrics_ops.so",
31+
":tracking_metrics_ops.so",
3132
],
3233
)
3334

@@ -56,6 +57,31 @@ cc_binary(
5657
],
5758
)
5859

60+
cc_binary(
61+
name = "tracking_metrics_ops.so",
62+
srcs = [
63+
"metrics_ops.cc",
64+
"tracking_metrics_ops.cc",
65+
],
66+
copts = [
67+
"-pthread",
68+
],
69+
linkshared = 1,
70+
deps = [
71+
":utils",
72+
"//waymo_open_dataset:label_cc_proto",
73+
"//waymo_open_dataset/metrics:config_util",
74+
"//waymo_open_dataset/metrics:tracking_metrics",
75+
"//waymo_open_dataset/protos:breakdown_cc_proto",
76+
"//waymo_open_dataset/protos:metrics_cc_proto",
77+
"@com_google_absl//absl/container:flat_hash_map",
78+
"@com_google_absl//absl/strings",
79+
"@com_google_absl//absl/types:optional",
80+
"@local_config_tf//:libtensorflow_framework",
81+
"@local_config_tf//:tf_header_lib",
82+
],
83+
)
84+
5985
# Tests
6086
py_test(
6187
name = "detection_metrics_ops_test",
@@ -71,3 +97,18 @@ py_test(
7197
"//waymo_open_dataset/protos:metrics_proto_py_pb2",
7298
],
7399
)
100+
101+
py_test(
102+
name = "tracking_metrics_ops_test",
103+
srcs = ["tracking_metrics_ops_test.py"],
104+
python_version = "PY3",
105+
deps = [
106+
":py_metrics_ops",
107+
# Implicit python proto dependency.
108+
# Implicit numpy dependency.
109+
# Implicit tensorflow dependency.
110+
"//waymo_open_dataset:label_proto_py_pb2",
111+
"//waymo_open_dataset/protos:breakdown_proto_py_pb2",
112+
"//waymo_open_dataset/protos:metrics_proto_py_pb2",
113+
],
114+
)

0 commit comments

Comments
 (0)