Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
BUILD 2.47 KB
Newer Older
Vivek Rathod's avatar
move data creation tools into a subfolder.  
Vivek Rathod committed
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84 
# Tensorflow Object Detection API: main runnables.

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

licenses(["notice"])

# Apache 2.0

py_binary(
    name = "create_kitti_tf_record",
    srcs = [
        "create_kitti_tf_record.py",
    ],
    deps = [
        "//third_party/py/PIL:pil",
        "//third_party/py/lxml",
        "//tensorflow",
        "//tensorflow_models/object_detection/utils:dataset_util",
        "//tensorflow_models/object_detection/utils:label_map_util",
        "//tensorflow_models/object_detection/utils:np_box_ops",
    ],
)

py_test(
    name = "create_kitti_tf_record_test",
    srcs = [
        "create_kitti_tf_record_test.py",
    ],
    deps = [
        ":create_kitti_tf_record",
        "//tensorflow",
    ],
)

py_binary(
    name = "create_pascal_tf_record",
    srcs = [
        "create_pascal_tf_record.py",
    ],
    deps = [
        "//third_party/py/PIL:pil",
        "//third_party/py/lxml",
        "//tensorflow",
        "//tensorflow_models/object_detection/utils:dataset_util",
        "//tensorflow_models/object_detection/utils:label_map_util",
    ],
)

py_test(
    name = "create_pascal_tf_record_test",
    srcs = [
        "create_pascal_tf_record_test.py",
    ],
    deps = [
        ":create_pascal_tf_record",
        "//tensorflow",
    ],
)

py_binary(
    name = "create_pet_tf_record",
    srcs = [
        "create_pet_tf_record.py",
    ],
    deps = [
        "//third_party/py/PIL:pil",
        "//third_party/py/lxml",
        "//tensorflow",
        "//tensorflow_models/object_detection/utils:dataset_util",
        "//tensorflow_models/object_detection/utils:label_map_util",
    ],
)

py_library(
    name = "oid_tfrecord_creation",
    srcs = ["oid_tfrecord_creation.py"],
    deps = [
        "//tensorflow",
        "//tensorflow_models/object_detection/core:standard_fields",
        "//tensorflow_models/object_detection/utils:dataset_util",
    ],
)
Vivek Rathod's avatar
Add dataset tools, pretrained models and documentatinos for open image dataset.  
Vivek Rathod committed
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107 

py_test(
    name = "oid_tfrecord_creation_test",
    srcs = ["oid_tfrecord_creation_test.py"],
    deps = [
        ":oid_tfrecord_creation",
        "//third_party/py/contextlib2",
        "//third_party/py/pandas",
        "//third_party/py/tensorflow",
    ],
)

py_binary(
    name = "create_oid_tf_record",
    srcs = ["create_oid_tf_record.py"],
    deps = [
        ":oid_tfrecord_creation",
        "//third_party/py/contextlib2",
        "//third_party/py/pandas",
        "//tensorflow",
        "//tensorflow_models/object_detection/utils:label_map_util",
    ],
)