BUILD 23.8 KB
Newer Older
1
# Description:
2
#   Contains files for loading, training and evaluating TF-Slim-based models.
3
# load("//devtools/python/blaze:python3.bzl", "py2and3_test")
4
load("//devtools/python/blaze:pytype.bzl", "pytype_strict_binary")
5

6
7
8
package(
    default_visibility = ["//visibility:public"],
)
9
10
11
12
13
14
15
16

licenses(["notice"])  # Apache 2.0

exports_files(["LICENSE"])

py_library(
    name = "dataset_utils",
    srcs = ["datasets/dataset_utils.py"],
Derek Chow's avatar
Derek Chow committed
17
    deps = [
18
        "//third_party/py/six",
19
        # "//tensorflow",
Derek Chow's avatar
Derek Chow committed
20
    ],
21
22
)

derekjchow's avatar
derekjchow committed
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
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"],
40
    python_version = "PY3",
derekjchow's avatar
derekjchow committed
41
    deps = [
42
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
43
        # "//numpy",
44
        "//third_party/py/six",
45
        # "//tensorflow",
derekjchow's avatar
derekjchow committed
46
47
48
    ],
)

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

60
py_library(
61
62
    name = "download_and_convert_flowers",
    srcs = ["datasets/download_and_convert_flowers.py"],
63
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
64
65
    deps = [
        ":dataset_utils",
66
        "//third_party/py/six",
67
        # "//tensorflow",
Derek Chow's avatar
Derek Chow committed
68
    ],
69
70
)

71
py_library(
72
73
    name = "download_and_convert_mnist",
    srcs = ["datasets/download_and_convert_mnist.py"],
Derek Chow's avatar
Derek Chow committed
74
75
    deps = [
        ":dataset_utils",
76
        # "//numpy",
77
        "//third_party/py/six",
78
        # "//tensorflow",
Derek Chow's avatar
Derek Chow committed
79
    ],
80
81
)

82
83
84
py_library(
    name = "download_and_convert_visualwakewords_lib",
    srcs = ["datasets/download_and_convert_visualwakewords_lib.py"],
85
    srcs_version = "PY2AND3",
86
    deps = [
87
88
89
        ":dataset_utils",
        "//third_party/py/PIL:pil",
        "//third_party/py/contextlib2",
90
        "//third_party/py/six",
91
        # "//tensorflow",
92
93
94
    ],
)

95
96
97
py_library(
    name = "download_and_convert_visualwakewords",
    srcs = ["datasets/download_and_convert_visualwakewords.py"],
98
    srcs_version = "PY2AND3",
99
    deps = [
100
        ":download_and_convert_visualwakewords_lib",
101
102
103
104
        # "//tensorflow",
    ],
)

105
106
107
py_binary(
    name = "download_and_convert_data",
    srcs = ["download_and_convert_data.py"],
108
    python_version = "PY3",
109
    deps = [
110
111
112
113
        ":download_and_convert_cifar10",
        ":download_and_convert_flowers",
        ":download_and_convert_mnist",
        ":download_and_convert_visualwakewords",
114
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
115
116
117
118
        # "//tensorflow",
    ],
)

119
py_library(
120
121
    name = "cifar10",
    srcs = ["datasets/cifar10.py"],
Derek Chow's avatar
Derek Chow committed
122
123
    deps = [
        ":dataset_utils",
124
        # "//tensorflow",
125
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
126
    ],
127
128
)

129
py_library(
130
131
    name = "flowers",
    srcs = ["datasets/flowers.py"],
Derek Chow's avatar
Derek Chow committed
132
133
    deps = [
        ":dataset_utils",
134
        # "//tensorflow",
135
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
136
    ],
137
138
)

139
py_library(
140
141
    name = "imagenet",
    srcs = ["datasets/imagenet.py"],
Derek Chow's avatar
Derek Chow committed
142
143
    deps = [
        ":dataset_utils",
144
        "//third_party/py/six",
145
        # "//tensorflow",
146
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
147
    ],
148
149
)

150
py_library(
151
152
    name = "mnist",
    srcs = ["datasets/mnist.py"],
Derek Chow's avatar
Derek Chow committed
153
154
    deps = [
        ":dataset_utils",
155
        # "//tensorflow",
156
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
157
    ],
158
159
)

160
161
162
163
164
165
py_library(
    name = "visualwakewords",
    srcs = ["datasets/visualwakewords.py"],
    deps = [
        ":dataset_utils",
        # "//tensorflow",
166
        "//third_party/py/tf_slim:slim",
167
168
169
    ],
)

170
171
172
173
174
175
176
177
py_library(
    name = "dataset_factory",
    srcs = ["datasets/dataset_factory.py"],
    deps = [
        ":cifar10",
        ":flowers",
        ":imagenet",
        ":mnist",
178
        ":visualwakewords",
179
180
181
182
183
    ],
)

py_library(
    name = "model_deploy",
184
    srcs = ["deployment/model_deploy.py"],
Derek Chow's avatar
Derek Chow committed
185
    deps = [
186
        # "//tensorflow",
187
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
188
    ],
189
190
)

191
py_test(  # py2and3_test
192
    name = "model_deploy_test",
193
    srcs = ["deployment/model_deploy_test.py"],
194
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
195
196
    deps = [
        ":model_deploy",
197
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
198
        # "//numpy",
199
        # "//tensorflow",
200
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
201
    ],
202
203
204
)

py_library(
205
206
    name = "cifarnet_preprocessing",
    srcs = ["preprocessing/cifarnet_preprocessing.py"],
Derek Chow's avatar
Derek Chow committed
207
    deps = [
208
        # "//tensorflow",
209
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
210
    ],
211
212
213
214
)

py_library(
    name = "inception_preprocessing",
215
    srcs = ["preprocessing/inception_preprocessing.py"],
Derek Chow's avatar
Derek Chow committed
216
    deps = [
217
218
        # "//tensorflow",
        # "//tensorflow/python:control_flow_ops",
Derek Chow's avatar
Derek Chow committed
219
    ],
220
221
222
223
)

py_library(
    name = "lenet_preprocessing",
224
    srcs = ["preprocessing/lenet_preprocessing.py"],
Derek Chow's avatar
Derek Chow committed
225
    deps = [
226
        # "//tensorflow",
227
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
228
    ],
229
230
231
232
)

py_library(
    name = "vgg_preprocessing",
233
    srcs = ["preprocessing/vgg_preprocessing.py"],
Derek Chow's avatar
Derek Chow committed
234
    deps = [
235
        # "//tensorflow",
236
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
237
    ],
238
239
240
241
)

py_library(
    name = "preprocessing_factory",
242
    srcs = ["preprocessing/preprocessing_factory.py"],
243
    deps = [
244
        ":cifarnet_preprocessing",
245
246
247
        ":inception_preprocessing",
        ":lenet_preprocessing",
        ":vgg_preprocessing",
248
        "//third_party/py/tf_slim:slim",
249
250
251
    ],
)

252
253
254
255
256
257
258
# Typical networks definitions.

py_library(
    name = "nets",
    deps = [
        ":alexnet",
        ":cifarnet",
259
        ":cyclegan",
260
        ":i3d",
261
262
        ":inception",
        ":lenet",
263
        ":mobilenet",
264
        ":nasnet",
265
        ":overfeat",
266
        ":pix2pix",
maximneumann's avatar
maximneumann committed
267
        ":pnasnet",
268
269
        ":resnet_v1",
        ":resnet_v2",
270
        ":s3dg",
271
272
273
274
275
276
277
278
        ":vgg",
    ],
)

py_library(
    name = "alexnet",
    srcs = ["nets/alexnet.py"],
    srcs_version = "PY2AND3",
279
    deps = [
pkulzc's avatar
pkulzc committed
280
        # "//tensorflow",
281
        "//third_party/py/tf_slim:slim",
282
    ],
283
284
)

285
py_test(  # py2and3_test
286
287
288
289
    name = "alexnet_test",
    size = "medium",
    srcs = ["nets/alexnet_test.py"],
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
290
291
    deps = [
        ":alexnet",
292
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
293
        # "//tensorflow",
294
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
295
    ],
296
297
298
299
300
)

py_library(
    name = "cifarnet",
    srcs = ["nets/cifarnet.py"],
Derek Chow's avatar
Derek Chow committed
301
    deps = [
302
        # "//tensorflow",
303
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
304
    ],
305
306
)

307
308
309
310
311
py_library(
    name = "cyclegan",
    srcs = ["nets/cyclegan.py"],
    deps = [
        # "//numpy",
312
        "//third_party/py/six",
313
        # "//tensorflow",
314
315
        "//third_party/py/tf_slim:slim",
        # "//tensorflow/python:tensor_util",
316
317
318
    ],
)

319
py_test(  # py2and3_test
320
321
322
323
324
325
    name = "cyclegan_test",
    srcs = ["nets/cyclegan_test.py"],
    shard_count = 3,
    srcs_version = "PY2AND3",
    deps = [
        ":cyclegan",
326
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
327
        # "//tensorflow",
328
329
330
331
332
333
334
    ],
)

py_library(
    name = "dcgan",
    srcs = ["nets/dcgan.py"],
    deps = [
335
        "//third_party/py/six",
336
        # "//tensorflow",
337
        "//third_party/py/tf_slim:slim",
338
339
340
    ],
)

341
py_test(  # py2and3_test
342
343
344
345
346
347
    name = "dcgan_test",
    srcs = ["nets/dcgan_test.py"],
    shard_count = 3,
    srcs_version = "PY2AND3",
    deps = [
        ":dcgan",
348
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
349
        "//third_party/py/six",
350
        # "//tensorflow",
351
352
353
    ],
)

354
355
356
357
358
359
360
361
py_library(
    name = "i3d",
    srcs = ["nets/i3d.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":i3d_utils",
        ":s3dg",
        # "//tensorflow",
362
        "//third_party/py/tf_slim:slim",
363
364
365
    ],
)

366
py_test(  # py2and3_test
367
368
369
370
371
372
373
    name = "i3d_test",
    size = "large",
    srcs = ["nets/i3d_test.py"],
    shard_count = 3,
    srcs_version = "PY2AND3",
    deps = [
        ":i3d",
374
375
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
        "//third_party/py/six",
376
        # "//tensorflow",
377
        "//third_party/py/tf_slim:slim",
378
379
380
381
382
383
384
385
    ],
)

py_library(
    name = "i3d_utils",
    srcs = ["nets/i3d_utils.py"],
    srcs_version = "PY2AND3",
    deps = [
386
        # "//numpy",
387
        # "//tensorflow",
388
        "//third_party/py/tf_slim:slim",
389
390
391
    ],
)

392
393
394
395
396
397
398
399
400
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
401
        ":inception_v4",
402
403
404
    ],
)

Alex Kurakin's avatar
Alex Kurakin committed
405
406
407
408
py_library(
    name = "inception_utils",
    srcs = ["nets/inception_utils.py"],
    srcs_version = "PY2AND3",
409
    deps = [
pkulzc's avatar
pkulzc committed
410
        # "//tensorflow",
411
        "//third_party/py/tf_slim:slim",
412
    ],
Alex Kurakin's avatar
Alex Kurakin committed
413
414
)

415
416
417
418
py_library(
    name = "inception_v1",
    srcs = ["nets/inception_v1.py"],
    srcs_version = "PY2AND3",
Alex Kurakin's avatar
Alex Kurakin committed
419
420
    deps = [
        ":inception_utils",
421
        # "//tensorflow",
422
        "//third_party/py/tf_slim:slim",
Alex Kurakin's avatar
Alex Kurakin committed
423
    ],
424
425
426
427
428
429
)

py_library(
    name = "inception_v2",
    srcs = ["nets/inception_v2.py"],
    srcs_version = "PY2AND3",
Alex Kurakin's avatar
Alex Kurakin committed
430
431
    deps = [
        ":inception_utils",
432
        # "//tensorflow",
433
        "//third_party/py/tf_slim:slim",
Alex Kurakin's avatar
Alex Kurakin committed
434
    ],
435
436
437
438
439
440
)

py_library(
    name = "inception_v3",
    srcs = ["nets/inception_v3.py"],
    srcs_version = "PY2AND3",
Alex Kurakin's avatar
Alex Kurakin committed
441
442
    deps = [
        ":inception_utils",
443
        # "//tensorflow",
444
        "//third_party/py/tf_slim:slim",
Alex Kurakin's avatar
Alex Kurakin committed
445
446
447
448
449
450
451
452
453
    ],
)

py_library(
    name = "inception_v4",
    srcs = ["nets/inception_v4.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":inception_utils",
454
        # "//tensorflow",
455
        "//third_party/py/tf_slim:slim",
Alex Kurakin's avatar
Alex Kurakin committed
456
    ],
457
458
459
460
461
462
)

py_library(
    name = "inception_resnet_v2",
    srcs = ["nets/inception_resnet_v2.py"],
    srcs_version = "PY2AND3",
463
    deps = [
pkulzc's avatar
pkulzc committed
464
        # "//tensorflow",
465
        "//third_party/py/tf_slim:slim",
466
    ],
467
468
)

469
py_test(  # py2and3_test
470
471
472
473
474
    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
475
476
    deps = [
        ":inception",
477
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
478
        # "//numpy",
479
        # "//tensorflow",
480
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
481
    ],
482
483
)

484
py_test(  # py2and3_test
485
486
487
488
489
    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
490
491
    deps = [
        ":inception",
492
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
493
        # "//numpy",
494
        # "//tensorflow",
495
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
496
    ],
497
498
)

499
py_test(  # py2and3_test
500
501
502
503
504
    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
505
506
    deps = [
        ":inception",
507
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
508
        # "//numpy",
509
        # "//tensorflow",
510
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
511
    ],
512
513
)

514
py_test(  # py2and3_test
Alex Kurakin's avatar
Alex Kurakin committed
515
516
517
518
519
    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
520
521
    deps = [
        ":inception",
522
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
523
        # "//tensorflow",
524
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
525
    ],
Alex Kurakin's avatar
Alex Kurakin committed
526
527
)

528
py_test(  # py2and3_test
529
530
531
    name = "inception_resnet_v2_test",
    size = "large",
    srcs = ["nets/inception_resnet_v2_test.py"],
532
    shard_count = 4,
533
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
534
535
    deps = [
        ":inception",
536
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
537
        # "//tensorflow",
538
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
539
    ],
540
541
)

542
543
544
py_library(
    name = "lenet",
    srcs = ["nets/lenet.py"],
Derek Chow's avatar
Derek Chow committed
545
    deps = [
546
        # "//tensorflow",
547
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
548
    ],
549
550
)

andrewghoward's avatar
andrewghoward committed
551
552
553
554
py_library(
    name = "mobilenet_v1",
    srcs = ["nets/mobilenet_v1.py"],
    srcs_version = "PY2AND3",
555
    deps = [
pkulzc's avatar
pkulzc committed
556
        # "//tensorflow",
557
        "//third_party/py/tf_slim:slim",
558
    ],
andrewghoward's avatar
andrewghoward committed
559
560
)

561
562
563
564
565
566
567
568
569
py_library(
    name = "mobilenet_common",
    srcs = [
        "nets/mobilenet/conv_blocks.py",
        "nets/mobilenet/mobilenet.py",
    ],
    srcs_version = "PY2AND3",
    deps = [
        # "//tensorflow",
570
        "//third_party/py/tf_slim:slim",
571
572
573
    ],
)

574
575
py_library(
    name = "mobilenet_v2",
576
    srcs = ["nets/mobilenet/mobilenet_v2.py"],
577
    srcs_version = "PY2AND3",
578
    deps = [
579
580
        ":mobilenet_common",
        # "//tensorflow",
581
        "//third_party/py/tf_slim:slim",
582
583
584
585
586
587
588
589
590
    ],
)

py_library(
    name = "mobilenet_v3",
    srcs = ["nets/mobilenet/mobilenet_v3.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":mobilenet_common",
591
        # "//numpy",
592
        # "//tensorflow",
593
        "//third_party/py/tf_slim:slim",
594
595
596
    ],
)

597
py_test(  # py2and3_test
598
599
600
601
602
    name = "mobilenet_v2_test",
    srcs = ["nets/mobilenet/mobilenet_v2_test.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":mobilenet",
603
        ":mobilenet_common",
604
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
605
        "//third_party/py/six",
606
        # "//tensorflow",
607
        "//third_party/py/tf_slim:slim",
608
609
610
    ],
)

611
612
613
py_test(  # py2and3_test
    name = "mobilenet_v3_test",
    srcs = ["nets/mobilenet/mobilenet_v3_test.py"],
614
    shard_count = 2,
615
616
617
    srcs_version = "PY2AND3",
    deps = [
        ":mobilenet",
618
619
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
        "//testing/pybase:parameterized",
620
        "//third_party/py/absl/testing:absltest",
621
622
623
624
        # "//tensorflow",
    ],
)

625
626
py_library(
    name = "mobilenet",
627
628
    deps = [
        ":mobilenet_v1",
629
        ":mobilenet_v2",
630
        ":mobilenet_v3",
631
632
633
    ],
)

634
py_test(  # py2and3_test
andrewghoward's avatar
andrewghoward committed
635
636
637
638
639
640
641
    name = "mobilenet_v1_test",
    size = "large",
    srcs = ["nets/mobilenet_v1_test.py"],
    shard_count = 3,
    srcs_version = "PY2AND3",
    deps = [
        ":mobilenet_v1",
642
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
643
        # "//numpy",
644
        # "//tensorflow",
645
        "//third_party/py/tf_slim:slim",
646
647
648
    ],
)

649
650
651
py_binary(
    name = "mobilenet_v1_train",
    srcs = ["nets/mobilenet_v1_train.py"],
652
    python_version = "PY3",
653
654
655
656
    deps = [
        ":dataset_factory",
        ":mobilenet_v1",
        ":preprocessing_factory",
657
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
658
        # "//tensorflow",
659
        "//third_party/py/tf_slim:slim",
660
        # "//tensorflow/contrib/quantize:quantize_graph",
661
662
663
664
665
666
    ],
)

py_binary(
    name = "mobilenet_v1_eval",
    srcs = ["nets/mobilenet_v1_eval.py"],
667
668
    python_version = "PY3",
    srcs_version = "PY3",
669
670
671
672
    deps = [
        ":dataset_factory",
        ":mobilenet_v1",
        ":preprocessing_factory",
673
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
674
        # "//tensorflow",
675
        "//third_party/py/tf_slim:slim",
676
        # "//tensorflow/contrib/quantize:quantize_graph",
677
678
679
    ],
)

680
681
682
683
684
py_library(
    name = "nasnet_utils",
    srcs = ["nets/nasnet/nasnet_utils.py"],
    srcs_version = "PY2AND3",
    deps = [
685
        # "//tensorflow",
686
        "//third_party/py/tf_slim:slim",
687
688
689
690
691
692
693
694
695
    ],
)

py_library(
    name = "nasnet",
    srcs = ["nets/nasnet/nasnet.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":nasnet_utils",
696
        # "//tensorflow",
697
        "//third_party/py/tf_slim:slim",
698
        # "//tensorflow/contrib/training:training_py",
699
700
701
    ],
)

702
py_test(  # py2and3_test
703
704
705
706
707
708
    name = "nasnet_utils_test",
    size = "medium",
    srcs = ["nets/nasnet/nasnet_utils_test.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":nasnet_utils",
709
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
710
        # "//tensorflow",
711
712
713
    ],
)

714
py_test(  # py2and3_test
715
716
717
718
719
720
721
    name = "nasnet_test",
    size = "large",
    srcs = ["nets/nasnet/nasnet_test.py"],
    shard_count = 10,
    srcs_version = "PY2AND3",
    deps = [
        ":nasnet",
722
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
723
        # "//tensorflow",
724
        "//third_party/py/tf_slim:slim",
andrewghoward's avatar
andrewghoward committed
725
726
727
    ],
)

maximneumann's avatar
maximneumann committed
728
729
730
731
732
733
734
735
py_library(
    name = "pnasnet",
    srcs = ["nets/nasnet/pnasnet.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":nasnet",
        ":nasnet_utils",
        # "//tensorflow",
736
        "//third_party/py/tf_slim:slim",
737
        # "//tensorflow/contrib/training:training_py",
maximneumann's avatar
maximneumann committed
738
739
740
    ],
)

741
py_test(  # py2and3_test
maximneumann's avatar
maximneumann committed
742
743
744
745
746
747
748
    name = "pnasnet_test",
    size = "large",
    srcs = ["nets/nasnet/pnasnet_test.py"],
    shard_count = 4,
    srcs_version = "PY2AND3",
    deps = [
        ":pnasnet",
749
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
maximneumann's avatar
maximneumann committed
750
        # "//tensorflow",
751
        "//third_party/py/tf_slim:slim",
maximneumann's avatar
maximneumann committed
752
753
754
    ],
)

755
py_library(
756
757
758
    name = "overfeat",
    srcs = ["nets/overfeat.py"],
    srcs_version = "PY2AND3",
759
    deps = [
pkulzc's avatar
pkulzc committed
760
        # "//tensorflow",
761
        "//third_party/py/tf_slim:slim",
762
    ],
763
764
)

765
py_test(  # py2and3_test
766
767
768
769
    name = "overfeat_test",
    size = "medium",
    srcs = ["nets/overfeat_test.py"],
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
770
771
    deps = [
        ":overfeat",
772
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
773
        # "//tensorflow",
774
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
775
    ],
776
777
)

778
779
780
781
py_library(
    name = "pix2pix",
    srcs = ["nets/pix2pix.py"],
    srcs_version = "PY2AND3",
782
    deps = [
pkulzc's avatar
pkulzc committed
783
        # "//tensorflow",
784
        "//third_party/py/tf_slim:slim",
785
    ],
786
787
)

788
py_test(  # py2and3_test
789
790
791
792
793
    name = "pix2pix_test",
    srcs = ["nets/pix2pix_test.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":pix2pix",
794
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
795
        # "//tensorflow",
796
        "//third_party/py/tf_slim:slim",
797
798
799
    ],
)

800
801
802
803
py_library(
    name = "resnet_utils",
    srcs = ["nets/resnet_utils.py"],
    srcs_version = "PY2AND3",
804
    deps = [
pkulzc's avatar
pkulzc committed
805
        # "//tensorflow",
806
        "//third_party/py/tf_slim:slim",
807
    ],
808
809
810
811
812
813
814
815
)

py_library(
    name = "resnet_v1",
    srcs = ["nets/resnet_v1.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":resnet_utils",
816
        # "//tensorflow",
817
        "//third_party/py/tf_slim:slim",
818
819
820
    ],
)

821
py_test(  # py2and3_test
822
823
    name = "resnet_v1_test",
    size = "medium",
824
    timeout = "long",
825
    srcs = ["nets/resnet_v1_test.py"],
Derek Chow's avatar
Derek Chow committed
826
    shard_count = 2,
827
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
828
    deps = [
829
        ":resnet_utils",
Derek Chow's avatar
Derek Chow committed
830
        ":resnet_v1",
831
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
832
        # "//numpy",
833
        # "//tensorflow",
834
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
835
    ],
836
837
838
839
840
841
842
843
)

py_library(
    name = "resnet_v2",
    srcs = ["nets/resnet_v2.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":resnet_utils",
844
        # "//tensorflow",
845
        "//third_party/py/tf_slim:slim",
846
847
848
    ],
)

849
py_test(  # py2and3_test
850
851
852
    name = "resnet_v2_test",
    size = "medium",
    srcs = ["nets/resnet_v2_test.py"],
Derek Chow's avatar
Derek Chow committed
853
    shard_count = 2,
854
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
855
    deps = [
856
        ":resnet_utils",
Derek Chow's avatar
Derek Chow committed
857
        ":resnet_v2",
858
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
859
        # "//numpy",
860
        # "//tensorflow",
861
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
862
    ],
863
864
)

865
866
867
868
869
870
871
py_library(
    name = "s3dg",
    srcs = ["nets/s3dg.py"],
    srcs_version = "PY2AND3",
    deps = [
        ":i3d_utils",
        # "//tensorflow",
872
        "//third_party/py/tf_slim:slim",
873
874
875
    ],
)

876
py_test(  # py2and3_test
877
878
879
880
881
882
883
    name = "s3dg_test",
    size = "large",
    srcs = ["nets/s3dg_test.py"],
    shard_count = 3,
    srcs_version = "PY2AND3",
    deps = [
        ":s3dg",
884
885
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
        "//third_party/py/six",
886
887
888
889
        # "//tensorflow",
    ],
)

890
891
892
893
py_library(
    name = "vgg",
    srcs = ["nets/vgg.py"],
    srcs_version = "PY2AND3",
894
    deps = [
pkulzc's avatar
pkulzc committed
895
        # "//tensorflow",
896
        "//third_party/py/tf_slim:slim",
897
    ],
898
899
)

900
py_test(  # py2and3_test
901
902
903
904
    name = "vgg_test",
    size = "medium",
    srcs = ["nets/vgg_test.py"],
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
905
906
    deps = [
        ":vgg",
907
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
908
        # "//tensorflow",
909
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
910
    ],
911
912
913
914
915
)

py_library(
    name = "nets_factory",
    srcs = ["nets/nets_factory.py"],
Derek Chow's avatar
Derek Chow committed
916
917
    deps = [
        ":nets",
918
        "//third_party/py/tf_slim:slim",
Derek Chow's avatar
Derek Chow committed
919
    ],
920
921
)

922
py_test(  # py2and3_test
923
    name = "nets_factory_test",
924
    size = "large",
925
    srcs = ["nets/nets_factory_test.py"],
926
    shard_count = 3,
927
    srcs_version = "PY2AND3",
Derek Chow's avatar
Derek Chow committed
928
929
    deps = [
        ":nets_factory",
930
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
931
        # "//tensorflow",
Derek Chow's avatar
Derek Chow committed
932
    ],
933
934
)

935
936
937
938
939
940
941
pytype_strict_binary(
    name = "post_training_quantization",
    srcs = ["nets/post_training_quantization.py"],
    python_version = "PY3",
    deps = [
        ":nets_factory",
        ":preprocessing_factory",
942
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
943
944
945
946
947
948
949
        "//third_party/py/absl:app",
        "//third_party/py/absl/flags",
        # "//tensorflow",
        # "//tensorflow_datasets",
    ],
)

950
951
py_library(
    name = "train_image_classifier_lib",
952
    srcs = ["train_image_classifier.py"],
953
    srcs_version = "PY2AND3",
954
955
956
957
958
    deps = [
        ":dataset_factory",
        ":model_deploy",
        ":nets_factory",
        ":preprocessing_factory",
959
        # "//tensorflow",
960
        "//third_party/py/tf_slim:slim",
961
        # "//tensorflow/contrib/quantize:quantize_graph",
962
    ],
963
964
)

965
966
967
py_binary(
    name = "train_image_classifier",
    srcs = ["train_image_classifier.py"],
968
969
    # WARNING: not supported in bazel; will be commented out by copybara.
    # paropts = ["--compress"],
970
    python_version = "PY3",
971
972
    deps = [
        ":train_image_classifier_lib",
973
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
974
975
976
    ],
)

977
978
py_library(
    name = "eval_image_classifier_lib",
979
    srcs = ["eval_image_classifier.py"],
980
981
    deps = [
        ":dataset_factory",
982
        ":nets_factory",
983
        ":preprocessing_factory",
984
        # "//tensorflow",
985
        "//third_party/py/tf_slim:slim",
986
        # "//tensorflow/contrib/quantize:quantize_graph",
987
988
    ],
)
989

990
991
992
py_binary(
    name = "eval_image_classifier",
    srcs = ["eval_image_classifier.py"],
993
    python_version = "PY3",
994
995
    deps = [
        ":eval_image_classifier_lib",
996
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
997
998
999
    ],
)

1000
1001
1002
py_binary(
    name = "export_inference_graph",
    srcs = ["export_inference_graph.py"],
1003
1004
    # WARNING: not supported in bazel; will be commented out by copybara.
    # paropts = ["--compress"],
1005
1006
1007
1008
1009
    python_version = "PY3",
    deps = [
        ":export_inference_graph_lib",
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
    ],
1010
1011
1012
1013
1014
)

py_library(
    name = "export_inference_graph_lib",
    srcs = ["export_inference_graph.py"],
1015
1016
1017
    deps = [
        ":dataset_factory",
        ":nets_factory",
1018
        # "//tensorflow",
1019
        # "//tensorflow/contrib/quantize:quantize_graph",
1020
        # "//tensorflow/python:platform",
1021
1022
1023
    ],
)

1024
py_test(  # py2and3_test
1025
1026
1027
    name = "export_inference_graph_test",
    size = "medium",
    srcs = ["export_inference_graph_test.py"],
1028
    python_version = "PY3",
1029
1030
1031
1032
1033
    srcs_version = "PY2AND3",
    tags = [
        "manual",
    ],
    deps = [
1034
        ":export_inference_graph_lib",
1035
        "//learning/brain/public:disable_tf2",  # build_cleaner: keep; go/disable_tf2
1036
1037
        # "//tensorflow",
        # "//tensorflow/python:platform",
1038
1039
    ],
)