waymo-open-dataset/src/waymo_open_dataset/wdl_limited/camera_segmentation/BUILD at master · Ravenkao/waymo-open-dataset · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
load("@wod_deps//:requirements.bzl", "requirement")

package(
    default_applicable_licenses = ["//waymo_open_dataset:license"],
    default_visibility = [
        "//visibility:public",
    ],
)

licenses(["notice"])

# Libs
py_library(
    name = "camera_segmentation_metrics",
    srcs = ["camera_segmentation_metrics.py"],
    deps = [
        requirement('numpy'),
        "//waymo_open_dataset/protos:camera_segmentation_metrics_proto_py_pb2",
        "@deeplab2",
    ],
)

# Tests
py_test(
    name = "camera_segmentation_metrics_test",
    srcs = ["camera_segmentation_metrics_test.py"],
    deps = [
        requirement('numpy'),
        requirement('tensorflow'),
        ":camera_segmentation_metrics",
    ],
)