BUILD 9.62 KB
Newer Older
1
# Description:
2
#   Contains files for loading, training and evaluating TF-Slim-based models.
3

4
package(default_visibility = ["//visibility:public"])
5
6
7
8
9

licenses(["notice"])  # Apache 2.0

exports_files(["LICENSE"])

10
package_group(name = "internal")
11
12
13
14

py_library(
    name = "dataset_utils",
    srcs = ["datasets/dataset_utils.py"],
Derek Chow's avatar
Derek Chow committed
15
16
17
    deps = [
        "//tensorflow",
    ],
18
19
)

derekjchow's avatar
derekjchow committed
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
sh_binary(
    name = "download_and_convert_imagenet",
    srcs = ["datasets/download_and_convert_imagenet.sh"],
    data = [
        "datasets/download_imagenet.sh",
        "datasets/imagenet_2012_validation_synset_labels.txt",
        "datasets/imagenet_lsvrc_2015_synsets.txt",
        "datasets/imagenet_metadata.txt",
        "datasets/preprocess_imagenet_validation_data.py",
        "datasets/process_bounding_boxes.py",
        ":build_imagenet_data",
    ],
)

py_binary(
    name = "build_imagenet_data",
    srcs = ["datasets/build_imagenet_data.py"],
    deps = [
        "//tensorflow",
    ],
)

42
py_library(
43
44
    name = "download_and_convert_cifar10",
    srcs = ["datasets/download_and_convert_cifar10.py"],
Derek Chow's avatar
Derek Chow committed
45
46
47
48
    deps = [
        ":dataset_utils",
        "//tensorflow",
    ],
49
50
)

51
py_library(
52
53
    name = "download_and_convert_flowers",
    srcs = ["datasets/download_and_convert_flowers.py"],
Derek Chow's avatar
Derek Chow committed
54
55
56
57
    deps = [
        ":dataset_utils",
        "//tensorflow",
    ],
58
59
)

60
py_library(
61
62
    name = "download_and_convert_mnist",
    srcs = ["datasets/download_and_convert_mnist.py"],
Derek Chow's avatar
Derek Chow committed
63
64
65
66
    deps = [
        ":dataset_utils",
        "//tensorflow",
    ],
67
68
)

69
70
71
72
73
74
75
76
77
78
py_binary(
    name = "download_and_convert_data",
    srcs = ["download_and_convert_data.py"],
    deps = [
        ":download_and_convert_cifar10",
        ":download_and_convert_flowers",
        ":download_and_convert_mnist",
    ],
)

79
80
81
py_binary(
    name = "cifar10",
    srcs = ["datasets/cifar10.py"],
Derek Chow's avatar
Derek Chow committed
82
83
84
85
    deps = [
        ":dataset_utils",
        "//tensorflow",
    ],
86
87
88
89
90
)

py_binary(
    name = "flowers",
    srcs = ["datasets/flowers.py"],
Derek Chow's avatar
Derek Chow committed
91
92
93
94
    deps = [
        ":dataset_utils",
        "//tensorflow",
    ],
95
96
97
98
99
)

py_binary(
    name = "imagenet",
    srcs = ["datasets/imagenet.py"],
Derek Chow's avatar
Derek Chow committed
100
101
102
103
    deps = [
        ":dataset_utils",
        "//tensorflow",
    ],
104
105
106
107
108
)

py_binary(
    name = "mnist",
    srcs = ["datasets/mnist.py"],
Derek Chow's avatar
Derek Chow committed
109
110
111
112
    deps = [
        ":dataset_utils",
        "//tensorflow",
    ],
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
)

py_library(
    name = "dataset_factory",
    srcs = ["datasets/dataset_factory.py"],
    deps = [
        ":cifar10",
        ":flowers",
        ":imagenet",
        ":mnist",
    ],
)

py_library(
    name = "model_deploy",
128
    srcs = ["deployment/model_deploy.py"],
Derek Chow's avatar
Derek Chow committed
129
130
131
    deps = [
        "//tensorflow",
    ],
132
133
134
135
)

py_test(
    name = "model_deploy_test",
136
    srcs = ["deployment/model_deploy_test.py"],
137
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
138
139
140
141
    deps = [
        ":model_deploy",
        "//tensorflow",
    ],
142
143
144
)

py_library(
145
146
    name = "cifarnet_preprocessing",
    srcs = ["preprocessing/cifarnet_preprocessing.py"],
Derek Chow's avatar
Derek Chow committed
147
148
149
    deps = [
        "//tensorflow",
    ],
150
151
152
153
)

py_library(
    name = "inception_preprocessing",
154
    srcs = ["preprocessing/inception_preprocessing.py"],
Derek Chow's avatar
Derek Chow committed
155
156
157
    deps = [
        "//tensorflow",
    ],
158
159
160
161
)

py_library(
    name = "lenet_preprocessing",
162
    srcs = ["preprocessing/lenet_preprocessing.py"],
Derek Chow's avatar
Derek Chow committed
163
164
165
    deps = [
        "//tensorflow",
    ],
166
167
168
169
)

py_library(
    name = "vgg_preprocessing",
170
    srcs = ["preprocessing/vgg_preprocessing.py"],
Derek Chow's avatar
Derek Chow committed
171
172
173
    deps = [
        "//tensorflow",
    ],
174
175
176
177
)

py_library(
    name = "preprocessing_factory",
178
    srcs = ["preprocessing/preprocessing_factory.py"],
179
    deps = [
180
        ":cifarnet_preprocessing",
181
182
183
        ":inception_preprocessing",
        ":lenet_preprocessing",
        ":vgg_preprocessing",
Derek Chow's avatar
Derek Chow committed
184
        "//tensorflow",
185
186
187
    ],
)

188
189
190
191
192
193
194
195
196
# Typical networks definitions.

py_library(
    name = "nets",
    deps = [
        ":alexnet",
        ":cifarnet",
        ":inception",
        ":lenet",
andrewghoward's avatar
andrewghoward committed
197
        ":mobilenet_v1",
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
        ":overfeat",
        ":resnet_v1",
        ":resnet_v2",
        ":vgg",
    ],
)

py_library(
    name = "alexnet",
    srcs = ["nets/alexnet.py"],
    srcs_version = "PY2AND3",
)

py_test(
    name = "alexnet_test",
    size = "medium",
    srcs = ["nets/alexnet_test.py"],
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
216
217
218
219
    deps = [
        ":alexnet",
        "//tensorflow",
    ],
220
221
222
223
224
)

py_library(
    name = "cifarnet",
    srcs = ["nets/cifarnet.py"],
Derek Chow's avatar
Derek Chow committed
225
226
227
    deps = [
        "//tensorflow",
    ],
228
229
230
231
232
233
234
235
236
237
238
)

py_library(
    name = "inception",
    srcs = ["nets/inception.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":inception_resnet_v2",
        ":inception_v1",
        ":inception_v2",
        ":inception_v3",
Alex Kurakin's avatar
Alex Kurakin committed
239
        ":inception_v4",
240
241
242
    ],
)

Alex Kurakin's avatar
Alex Kurakin committed
243
244
245
246
247
248
py_library(
    name = "inception_utils",
    srcs = ["nets/inception_utils.py"],
    srcs_version = "PY2AND3",
)

249
250
251
252
py_library(
    name = "inception_v1",
    srcs = ["nets/inception_v1.py"],
    srcs_version = "PY2AND3",
Alex Kurakin's avatar
Alex Kurakin committed
253
254
255
    deps = [
        ":inception_utils",
    ],
256
257
258
259
260
261
)

py_library(
    name = "inception_v2",
    srcs = ["nets/inception_v2.py"],
    srcs_version = "PY2AND3",
Alex Kurakin's avatar
Alex Kurakin committed
262
263
264
    deps = [
        ":inception_utils",
    ],
265
266
267
268
269
270
)

py_library(
    name = "inception_v3",
    srcs = ["nets/inception_v3.py"],
    srcs_version = "PY2AND3",
Alex Kurakin's avatar
Alex Kurakin committed
271
272
273
274
275
276
277
278
279
280
281
282
    deps = [
        ":inception_utils",
    ],
)

py_library(
    name = "inception_v4",
    srcs = ["nets/inception_v4.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":inception_utils",
    ],
283
284
285
286
287
288
289
290
291
292
293
294
295
296
)

py_library(
    name = "inception_resnet_v2",
    srcs = ["nets/inception_resnet_v2.py"],
    srcs_version = "PY2AND3",
)

py_test(
    name = "inception_v1_test",
    size = "large",
    srcs = ["nets/inception_v1_test.py"],
    shard_count = 3,
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
297
298
299
300
    deps = [
        ":inception",
        "//tensorflow",
    ],
301
302
303
304
305
306
307
308
)

py_test(
    name = "inception_v2_test",
    size = "large",
    srcs = ["nets/inception_v2_test.py"],
    shard_count = 3,
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
309
310
311
312
    deps = [
        ":inception",
        "//tensorflow",
    ],
313
314
315
316
317
318
319
320
)

py_test(
    name = "inception_v3_test",
    size = "large",
    srcs = ["nets/inception_v3_test.py"],
    shard_count = 3,
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
321
322
323
324
    deps = [
        ":inception",
        "//tensorflow",
    ],
325
326
)

Alex Kurakin's avatar
Alex Kurakin committed
327
328
329
330
331
332
py_test(
    name = "inception_v4_test",
    size = "large",
    srcs = ["nets/inception_v4_test.py"],
    shard_count = 3,
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
333
334
335
336
    deps = [
        ":inception",
        "//tensorflow",
    ],
Alex Kurakin's avatar
Alex Kurakin committed
337
338
)

339
340
341
342
343
344
py_test(
    name = "inception_resnet_v2_test",
    size = "large",
    srcs = ["nets/inception_resnet_v2_test.py"],
    shard_count = 3,
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
345
346
347
348
    deps = [
        ":inception",
        "//tensorflow",
    ],
349
350
)

351
352
353
py_library(
    name = "lenet",
    srcs = ["nets/lenet.py"],
Derek Chow's avatar
Derek Chow committed
354
355
356
    deps = [
        "//tensorflow",
    ],
357
358
)

andrewghoward's avatar
andrewghoward committed
359
360
361
362
363
364
365
366
367
368
369
370
371
372
py_library(
    name = "mobilenet_v1",
    srcs = ["nets/mobilenet_v1.py"],
    srcs_version = "PY2AND3",
)

py_test(
    name = "mobilenet_v1_test",
    size = "large",
    srcs = ["nets/mobilenet_v1_test.py"],
    shard_count = 3,
    srcs_version = "PY2AND3",
    deps = [
        ":mobilenet_v1",
Derek Chow's avatar
Derek Chow committed
373
        "//tensorflow",
andrewghoward's avatar
andrewghoward committed
374
375
376
    ],
)

377
py_library(
378
379
380
381
382
383
384
385
386
387
    name = "overfeat",
    srcs = ["nets/overfeat.py"],
    srcs_version = "PY2AND3",
)

py_test(
    name = "overfeat_test",
    size = "medium",
    srcs = ["nets/overfeat_test.py"],
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
388
389
390
391
    deps = [
        ":overfeat",
        "//tensorflow",
    ],
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
)

py_library(
    name = "resnet_utils",
    srcs = ["nets/resnet_utils.py"],
    srcs_version = "PY2AND3",
)

py_library(
    name = "resnet_v1",
    srcs = ["nets/resnet_v1.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":resnet_utils",
    ],
)

py_test(
    name = "resnet_v1_test",
    size = "medium",
    srcs = ["nets/resnet_v1_test.py"],
Derek Chow's avatar
Derek Chow committed
413
    shard_count = 2,
414
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
415
416
417
418
    deps = [
        ":resnet_v1",
        "//tensorflow",
    ],
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
)

py_library(
    name = "resnet_v2",
    srcs = ["nets/resnet_v2.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":resnet_utils",
    ],
)

py_test(
    name = "resnet_v2_test",
    size = "medium",
    srcs = ["nets/resnet_v2_test.py"],
Derek Chow's avatar
Derek Chow committed
434
    shard_count = 2,
435
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
436
437
438
439
    deps = [
        ":resnet_v2",
        "//tensorflow",
    ],
440
441
442
443
444
445
446
447
448
449
450
451
452
)

py_library(
    name = "vgg",
    srcs = ["nets/vgg.py"],
    srcs_version = "PY2AND3",
)

py_test(
    name = "vgg_test",
    size = "medium",
    srcs = ["nets/vgg_test.py"],
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
453
454
455
456
    deps = [
        ":vgg",
        "//tensorflow",
    ],
457
458
459
460
461
)

py_library(
    name = "nets_factory",
    srcs = ["nets/nets_factory.py"],
Derek Chow's avatar
Derek Chow committed
462
463
464
465
    deps = [
        ":nets",
        "//tensorflow",
    ],
466
467
468
469
470
471
)

py_test(
    name = "nets_factory_test",
    size = "medium",
    srcs = ["nets/nets_factory_test.py"],
Derek Chow's avatar
Derek Chow committed
472
    shard_count = 2,
473
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
474
475
476
477
    deps = [
        ":nets_factory",
        "//tensorflow",
    ],
478
479
480
481
482
483
484
485
486
487
)

py_binary(
    name = "train_image_classifier",
    srcs = ["train_image_classifier.py"],
    deps = [
        ":dataset_factory",
        ":model_deploy",
        ":nets_factory",
        ":preprocessing_factory",
Derek Chow's avatar
Derek Chow committed
488
        "//tensorflow",
489
    ],
490
491
492
)

py_binary(
493
494
    name = "eval_image_classifier",
    srcs = ["eval_image_classifier.py"],
495
496
497
    deps = [
        ":dataset_factory",
        ":model_deploy",
498
        ":nets_factory",
499
        ":preprocessing_factory",
Derek Chow's avatar
Derek Chow committed
500
        "//tensorflow",
501
502
    ],
)
503
504
505
506
507
508
509

py_binary(
    name = "export_inference_graph",
    srcs = ["export_inference_graph.py"],
    deps = [
        ":dataset_factory",
        ":nets_factory",
Derek Chow's avatar
Derek Chow committed
510
        "//tensorflow",
511
512
513
514
515
516
517
518
519
520
521
522
523
524
    ],
)

py_test(
    name = "export_inference_graph_test",
    size = "medium",
    srcs = ["export_inference_graph_test.py"],
    srcs_version = "PY2AND3",
    tags = [
        "manual",
    ],
    deps = [
        ":export_inference_graph",
        ":nets_factory",
Derek Chow's avatar
Derek Chow committed
525
        "//tensorflow",
526
527
    ],
)