Path Lines of Code official/__init__.py 1 official/common/__init__.py 1 official/common/dataset_fn.py 7 official/common/distribute_utils.py 120 official/common/flags.py 64 official/common/registry_imports.py 4 official/common/streamz_counters.py 7 official/core/__init__.py 1 official/core/actions.py 125 official/core/base_task.py 136 official/core/base_trainer.py 289 official/core/config_definitions.py 85 official/core/exp_factory.py 8 official/core/export_base.py 77 official/core/input_reader.py 344 official/core/registry.py 42 official/core/task_factory.py 11 official/core/train_lib.py 95 official/core/train_utils.py 330 official/legacy/__init__.py 1 official/legacy/albert/__init__.py 1 official/legacy/albert/configs.py 14 official/legacy/bert/__init__.py 1 official/legacy/bert/bert_models.py 250 official/legacy/bert/common_flags.py 94 official/legacy/bert/configs.py 48 official/legacy/bert/export_tfhub.py 88 official/legacy/bert/input_pipeline.py 217 official/legacy/bert/model_saving_utils.py 26 official/legacy/bert/model_training_utils.py 331 official/legacy/bert/run_classifier.py 383 official/legacy/bert/run_pretraining.py 157 official/legacy/bert/run_squad.py 93 official/legacy/bert/run_squad_helper.py 348 official/legacy/bert/serving.py 95 official/legacy/detection/__init__.py 1 official/legacy/detection/configs/__init__.py 1 official/legacy/detection/configs/base_config.py 104 official/legacy/detection/configs/factory.py 21 official/legacy/detection/configs/maskrcnn_config.py 93 official/legacy/detection/configs/olnmask_config.py 116 official/legacy/detection/configs/retinanet_config.py 36 official/legacy/detection/configs/shapemask_config.py 75 official/legacy/detection/dataloader/__init__.py 1 official/legacy/detection/dataloader/anchor.py 223 official/legacy/detection/dataloader/factory.py 115 official/legacy/detection/dataloader/input_reader.py 57 official/legacy/detection/dataloader/maskrcnn_parser.py 199 official/legacy/detection/dataloader/mode_keys.py 7 official/legacy/detection/dataloader/olnmask_parser.py 182 official/legacy/detection/dataloader/retinanet_parser.py 234 official/legacy/detection/dataloader/shapemask_parser.py 288 official/legacy/detection/dataloader/tf_example_decoder.py 102 official/legacy/detection/evaluation/__init__.py 1 official/legacy/detection/evaluation/coco_evaluator.py 388 official/legacy/detection/evaluation/coco_utils.py 212 official/legacy/detection/evaluation/factory.py 30 official/legacy/detection/executor/__init__.py 1 official/legacy/detection/executor/detection_executor.py 110 official/legacy/detection/executor/distributed_executor.py 449 official/legacy/detection/main.py 197 official/legacy/detection/modeling/__init__.py 1 official/legacy/detection/modeling/architecture/__init__.py 1 official/legacy/detection/modeling/architecture/factory.py 161 official/legacy/detection/modeling/architecture/fpn.py 86 official/legacy/detection/modeling/architecture/heads.py 799 official/legacy/detection/modeling/architecture/identity.py 6 official/legacy/detection/modeling/architecture/nn_blocks.py 219 official/legacy/detection/modeling/architecture/nn_ops.py 55 official/legacy/detection/modeling/architecture/resnet.py 186 official/legacy/detection/modeling/architecture/spinenet.py 414 official/legacy/detection/modeling/base_model.py 68 official/legacy/detection/modeling/checkpoint_utils.py 67 official/legacy/detection/modeling/factory.py 16 official/legacy/detection/modeling/learning_rates.py 63 official/legacy/detection/modeling/losses.py 318 official/legacy/detection/modeling/maskrcnn_model.py 268 official/legacy/detection/modeling/olnmask_model.py 350 official/legacy/detection/modeling/optimizers.py 25 official/legacy/detection/modeling/retinanet_model.py 113 official/legacy/detection/modeling/shapemask_model.py 237 official/legacy/detection/ops/__init__.py 1 official/legacy/detection/ops/nms.py 97 official/legacy/detection/ops/postprocess_ops.py 238 official/legacy/detection/ops/roi_ops.py 240 official/legacy/detection/ops/spatial_transform_ops.py 314 official/legacy/detection/ops/target_ops.py 245 official/legacy/detection/utils/__init__.py 1 official/legacy/detection/utils/box_utils.py 361 official/legacy/detection/utils/class_utils.py 16 official/legacy/detection/utils/dataloader_utils.py 17 official/legacy/detection/utils/input_utils.py 159 official/legacy/detection/utils/mask_utils.py 84 official/legacy/image_classification/__init__.py 1 official/legacy/image_classification/augment.py 502 official/legacy/image_classification/callbacks.py 156 official/legacy/image_classification/classifier_trainer.py 332 official/legacy/image_classification/configs/__init__.py 1 official/legacy/image_classification/configs/base_configs.py 88 official/legacy/image_classification/configs/configs.py 98 official/legacy/image_classification/dataset_factory.py 319 official/legacy/image_classification/efficientnet/__init__.py 1 official/legacy/image_classification/efficientnet/common_modules.py 66 official/legacy/image_classification/efficientnet/efficientnet_config.py 39 official/legacy/image_classification/efficientnet/efficientnet_model.py 328 official/legacy/image_classification/efficientnet/tfhub_export.py 35 official/legacy/image_classification/learning_rate.py 67 official/legacy/image_classification/mnist_main.py 112 official/legacy/image_classification/optimizer_factory.py 128 official/legacy/image_classification/preprocessing.py 183 official/legacy/image_classification/resnet/__init__.py 1 official/legacy/image_classification/resnet/common.py 282 official/legacy/image_classification/resnet/imagenet_preprocessing.py 213 official/legacy/image_classification/resnet/resnet_config.py 34 official/legacy/image_classification/resnet/resnet_ctl_imagenet_main.py 127 official/legacy/image_classification/resnet/resnet_model.py 219 official/legacy/image_classification/resnet/resnet_runnable.py 151 official/legacy/image_classification/resnet/tfhub_export.py 31 official/legacy/image_classification/vgg/__init__.py 1 official/legacy/image_classification/vgg/vgg_config.py 25 official/legacy/image_classification/vgg/vgg_model.py 214 official/legacy/transformer/__init__.py 1 official/legacy/transformer/attention_layer.py 97 official/legacy/transformer/beam_search_v1.py 27 official/legacy/transformer/compute_bleu.py 79 official/legacy/transformer/data_download.py 244 official/legacy/transformer/data_pipeline.py 136 official/legacy/transformer/embedding_layer.py 39 official/legacy/transformer/ffn_layer.py 28 official/legacy/transformer/metrics.py 81 official/legacy/transformer/misc.py 221 official/legacy/transformer/model_params.py 54 official/legacy/transformer/model_utils.py 37 official/legacy/transformer/optimizer.py 23 official/legacy/transformer/transformer.py 302 official/legacy/transformer/transformer_main.py 322 official/legacy/transformer/translate.py 111 official/legacy/transformer/utils/__init__.py 1 official/legacy/transformer/utils/metrics.py 219 official/legacy/transformer/utils/tokenizer.py 330 official/modeling/__init__.py 1 official/modeling/activations/__init__.py 6 official/modeling/activations/gelu.py 4 official/modeling/activations/relu.py 5 official/modeling/activations/sigmoid.py 5 official/modeling/activations/swish.py 14 official/modeling/fast_training/experimental/tf2_utils_2x_wide.py 60 official/modeling/fast_training/progressive/policies.py 97 official/modeling/fast_training/progressive/train.py 37 official/modeling/fast_training/progressive/train_lib.py 75 official/modeling/fast_training/progressive/trainer.py 153 official/modeling/fast_training/progressive/utils.py 19 official/modeling/grad_utils.py 78 official/modeling/hyperparams/__init__.py 3 official/modeling/hyperparams/base_config.py 184 official/modeling/hyperparams/oneof.py 23 official/modeling/hyperparams/params_dict.py 228 official/modeling/multitask/__init__.py 1 official/modeling/multitask/base_model.py 14 official/modeling/multitask/base_trainer.py 108 official/modeling/multitask/configs.py 40 official/modeling/multitask/evaluator.py 118 official/modeling/multitask/interleaving_trainer.py 65 official/modeling/multitask/multitask.py 93 official/modeling/multitask/task_sampler.py 73 official/modeling/multitask/train_lib.py 190 official/modeling/optimization/__init__.py 7 official/modeling/optimization/adafactor_optimizer.py 1 official/modeling/optimization/configs/__init__.py 1 official/modeling/optimization/configs/learning_rate_config.py 73 official/modeling/optimization/configs/optimization_config.py 46 official/modeling/optimization/configs/optimizer_config.py 103 official/modeling/optimization/ema_optimizer.py 141 official/modeling/optimization/lars_optimizer.py 109 official/modeling/optimization/lr_schedule.py 280 official/modeling/optimization/optimizer_factory.py 96 official/modeling/optimization/slide_optimizer.py 1 official/modeling/performance.py 29 official/modeling/tf_utils.py 108 official/nlp/__init__.py 1 official/nlp/configs/__init__.py 1 official/nlp/configs/bert.py 18 official/nlp/configs/electra.py 16 official/nlp/configs/encoders.py 429 official/nlp/configs/experiment_configs.py 4 official/nlp/configs/experiments/glue_mnli_matched.yaml 45 official/nlp/configs/experiments/glue_mnli_text.yaml 50 official/nlp/configs/experiments/squad_v1.yaml 50 official/nlp/configs/finetuning_experiments.py 150 official/nlp/configs/models/albert_base.yaml 16 official/nlp/configs/models/bert_en_uncased_base.yaml 16 official/nlp/configs/pretraining_experiments.py 58 official/nlp/configs/wmt_transformer_experiments.py 82 official/nlp/continuous_finetune_lib.py 121 official/nlp/data/__init__.py 1 official/nlp/data/classifier_data_lib.py 1108 official/nlp/data/create_finetuning_data.py 351 official/nlp/data/create_pretraining_data.py 396 official/nlp/data/create_xlnet_pretraining_data.py 488 official/nlp/data/data_loader.py 10 official/nlp/data/data_loader_factory.py 7 official/nlp/data/dual_encoder_dataloader.py 90 official/nlp/data/pretrain_dataloader.py 380 official/nlp/data/pretrain_dynamic_dataloader.py 143 official/nlp/data/question_answering_dataloader.py 70 official/nlp/data/sentence_prediction_dataloader.py 183 official/nlp/data/sentence_retrieval_lib.py 101 official/nlp/data/squad_lib.py 706 official/nlp/data/squad_lib_sp.py 753 official/nlp/data/tagging_data_lib.py 270 official/nlp/data/tagging_dataloader.py 50 official/nlp/data/train_sentencepiece.py 73 official/nlp/data/wmt_dataloader.py 165 official/nlp/finetuning/binary_helper.py 348 official/nlp/finetuning/glue/flags.py 103 official/nlp/finetuning/glue/run_glue.py 204 official/nlp/finetuning/superglue/flags.py 112 official/nlp/finetuning/superglue/run_superglue.py 164 official/nlp/metrics/__init__.py 1 official/nlp/metrics/bleu.py 93 official/nlp/modeling/__init__.py 5 official/nlp/modeling/layers/__init__.py 38 official/nlp/modeling/layers/attention.py 51 official/nlp/modeling/layers/bigbird_attention.py 324 official/nlp/modeling/layers/block_diag_feedforward.py 111 official/nlp/modeling/layers/cls_head.py 195 official/nlp/modeling/layers/gated_feedforward.py 144 official/nlp/modeling/layers/gaussian_process.py 236 official/nlp/modeling/layers/kernel_attention.py 209 official/nlp/modeling/layers/masked_lm.py 61 official/nlp/modeling/layers/masked_softmax.py 36 official/nlp/modeling/layers/mat_mul_with_margin.py 31 official/nlp/modeling/layers/mobile_bert_layers.py 365 official/nlp/modeling/layers/multi_channel_attention.py 91 official/nlp/modeling/layers/on_device_embedding.py 57 official/nlp/modeling/layers/position_embedding.py 170 official/nlp/modeling/layers/relative_attention.py 224 official/nlp/modeling/layers/reuse_attention.py 343 official/nlp/modeling/layers/reuse_transformer.py 253 official/nlp/modeling/layers/rezero_transformer.py 204 official/nlp/modeling/layers/self_attention_mask.py 19 official/nlp/modeling/layers/spectral_normalization.py 166 official/nlp/modeling/layers/talking_heads_attention.py 55 official/nlp/modeling/layers/text_layers.py 385 official/nlp/modeling/layers/tn_expand_condense.py 92 official/nlp/modeling/layers/tn_transformer_expand_condense.py 181 official/nlp/modeling/layers/transformer.py 329 official/nlp/modeling/layers/transformer_encoder_block.py 213 official/nlp/modeling/layers/transformer_scaffold.py 207 official/nlp/modeling/layers/transformer_xl.py 354 official/nlp/modeling/layers/util.py 29 official/nlp/modeling/losses/__init__.py 1 official/nlp/modeling/losses/weighted_sparse_categorical_crossentropy.py 27 official/nlp/modeling/models/__init__.py 12 official/nlp/modeling/models/bert_classifier.py 71 official/nlp/modeling/models/bert_pretrainer.py 159 official/nlp/modeling/models/bert_span_labeler.py 50 official/nlp/modeling/models/bert_token_classifier.py 60 official/nlp/modeling/models/dual_encoder.py 92 official/nlp/modeling/models/electra_pretrainer.py 182 official/nlp/modeling/models/seq2seq_transformer.py 366 official/nlp/modeling/models/t5.py 1061 official/nlp/modeling/models/xlnet.py 235 official/nlp/modeling/networks/__init__.py 11 official/nlp/modeling/networks/albert_encoder.py 122 official/nlp/modeling/networks/bert_encoder.py 373 official/nlp/modeling/networks/classification.py 50 official/nlp/modeling/networks/encoder_scaffold.py 243 official/nlp/modeling/networks/funnel_transformer.py 341 official/nlp/modeling/networks/mobile_bert_encoder.py 105 official/nlp/modeling/networks/packed_sequence_embedding.py 176 official/nlp/modeling/networks/span_labeling.py 183 official/nlp/modeling/networks/xlnet_base.py 408 official/nlp/modeling/ops/__init__.py 3 official/nlp/modeling/ops/beam_search.py 306 official/nlp/modeling/ops/decoding_module.py 126 official/nlp/modeling/ops/sampling_module.py 266 official/nlp/modeling/ops/segment_extractor.py 88 official/nlp/optimization.py 165 official/nlp/serving/__init__.py 1 official/nlp/serving/export_savedmodel.py 113 official/nlp/serving/export_savedmodel_util.py 26 official/nlp/serving/serving_modules.py 373 official/nlp/tasks/__init__.py 6 official/nlp/tasks/dual_encoder.py 140 official/nlp/tasks/electra_task.py 171 official/nlp/tasks/masked_lm.py 137 official/nlp/tasks/question_answering.py 379 official/nlp/tasks/sentence_prediction.py 236 official/nlp/tasks/tagging.py 159 official/nlp/tasks/translation.py 239 official/nlp/tasks/utils.py 31 official/nlp/tools/__init__.py 1 official/nlp/tools/export_tfhub.py 140 official/nlp/tools/export_tfhub_lib.py 247 official/nlp/tools/squad_evaluate_v1_1.py 55 official/nlp/tools/squad_evaluate_v2_0.py 181 official/nlp/tools/tf1_bert_checkpoint_converter_lib.py 125 official/nlp/tools/tf2_albert_encoder_checkpoint_converter.py 109 official/nlp/tools/tf2_bert_encoder_checkpoint_converter.py 96 official/nlp/tools/tokenization.py 313 official/nlp/train.py 48 official/nlp/xlnet/__init__.py 1 official/nlp/xlnet/classifier_utils.py 99 official/nlp/xlnet/common_flags.py 110 official/nlp/xlnet/data_utils.py 512 official/nlp/xlnet/optimization.py 69 official/nlp/xlnet/preprocess_classification_data.py 303 official/nlp/xlnet/preprocess_pretrain_data.py 705 official/nlp/xlnet/preprocess_squad_data.py 69 official/nlp/xlnet/preprocess_utils.py 77 official/nlp/xlnet/run_classifier.py 130 official/nlp/xlnet/run_pretrain.py 108 official/nlp/xlnet/run_squad.py 221 official/nlp/xlnet/squad_utils.py 715 official/nlp/xlnet/training_utils.py 195 official/nlp/xlnet/xlnet_config.py 82 official/nlp/xlnet/xlnet_modeling.py 915 official/pip_package/setup.py 61 official/projects/__init__.py 1 official/projects/assemblenet/experiment/ucf101_assemblenet_plus_tpu.yaml 53 official/projects/assemblenet/experiment/ucf101_assemblenet_tpu.yaml 52 official/projects/assemblenet/modeling/assemblenet.py 671 official/projects/assemblenet/modeling/assemblenet_plus.py 470 official/projects/assemblenet/modeling/rep_flow_2d_layer.py 276 official/projects/assemblenet/train.py 71 official/projects/basnet/configs/basnet.py 118 official/projects/basnet/configs/experiments/basnet_dut_gpu.yaml 10 official/projects/basnet/evaluation/metrics.py 171 official/projects/basnet/losses/basnet_losses.py 35 official/projects/basnet/modeling/basnet_model.py 288 official/projects/basnet/modeling/nn_blocks.py 158 official/projects/basnet/modeling/refunet.py 106 official/projects/basnet/serving/basnet.py 31 official/projects/basnet/serving/export_saved_model.py 59 official/projects/basnet/tasks/basnet.py 171 official/projects/basnet/train.py 10 official/projects/bigbird/__init__.py 1 official/projects/bigbird/encoder.py 146 official/projects/bigbird/experiment_configs.py 80 official/projects/bigbird/experiments/glue_mnli_matched.yaml 49 official/projects/bigbird/experiments/squad_v1.yaml 55 official/projects/bigbird/recompute_grad.py 123 official/projects/bigbird/recomputing_dropout.py 74 official/projects/bigbird/stateless_dropout.py 58 official/projects/detr/configs/detr.py 77 official/projects/detr/dataloaders/coco.py 117 official/projects/detr/modeling/detr.py 191 official/projects/detr/modeling/transformer.py 582 official/projects/detr/ops/matchers.py 192 official/projects/detr/optimization.py 99 official/projects/detr/tasks/detection.py 219 official/projects/detr/train.py 38 official/projects/edgetpu/nlp/__init__.py 1 official/projects/edgetpu/nlp/configs/__init__.py 1 official/projects/edgetpu/nlp/configs/params.py 74 official/projects/edgetpu/nlp/experiments/downstream_tasks/glue_mnli.yaml 44 official/projects/edgetpu/nlp/experiments/downstream_tasks/mobilebert_baseline.yaml 22 official/projects/edgetpu/nlp/experiments/downstream_tasks/mobilebert_edgetpu_m.yaml 22 official/projects/edgetpu/nlp/experiments/downstream_tasks/mobilebert_edgetpu_s.yaml 22 official/projects/edgetpu/nlp/experiments/downstream_tasks/mobilebert_edgetpu_xs.yaml 22 official/projects/edgetpu/nlp/experiments/downstream_tasks/squad_v1.yaml 49 official/projects/edgetpu/nlp/experiments/mobilebert_baseline.yaml 141 official/projects/edgetpu/nlp/experiments/mobilebert_edgetpu_m.yaml 141 official/projects/edgetpu/nlp/experiments/mobilebert_edgetpu_s.yaml 141 official/projects/edgetpu/nlp/experiments/mobilebert_edgetpu_xs.yaml 141 official/projects/edgetpu/nlp/mobilebert_edgetpu_trainer.py 382 official/projects/edgetpu/nlp/modeling/__init__.py 1 official/projects/edgetpu/nlp/modeling/edgetpu_layers.py 78 official/projects/edgetpu/nlp/modeling/encoder.py 115 official/projects/edgetpu/nlp/modeling/model_builder.py 58 official/projects/edgetpu/nlp/modeling/pretrainer.py 76 official/projects/edgetpu/nlp/run_mobilebert_edgetpu_train.py 77 official/projects/edgetpu/nlp/serving/__init__.py 1 official/projects/edgetpu/nlp/serving/export_tflite_squad.py 108 official/projects/edgetpu/nlp/utils/__init__.py 1 official/projects/edgetpu/nlp/utils/utils.py 49 official/projects/edgetpu/vision/__init__.py 1 official/projects/edgetpu/vision/configs/__init__.py 1 official/projects/edgetpu/vision/configs/mobilenet_edgetpu_config.py 142 official/projects/edgetpu/vision/configs/semantic_segmentation_config.py 183 official/projects/edgetpu/vision/configs/semantic_segmentation_searched_config.py 147 official/projects/edgetpu/vision/dataloaders/__init__.py 1 official/projects/edgetpu/vision/dataloaders/classification_input.py 88 official/projects/edgetpu/vision/modeling/__init__.py 1 official/projects/edgetpu/vision/modeling/backbones/__init__.py 1 official/projects/edgetpu/vision/modeling/backbones/mobilenet_edgetpu.py 70 official/projects/edgetpu/vision/modeling/common_modules.py 91 official/projects/edgetpu/vision/modeling/custom_layers.py 268 official/projects/edgetpu/vision/modeling/heads/__init__.py 1 official/projects/edgetpu/vision/modeling/heads/bifpn_head.py 557 official/projects/edgetpu/vision/modeling/mobilenet_edgetpu_v1_model.py 65 official/projects/edgetpu/vision/modeling/mobilenet_edgetpu_v1_model_blocks.py 289 official/projects/edgetpu/vision/modeling/mobilenet_edgetpu_v2_model.py 98 official/projects/edgetpu/vision/modeling/mobilenet_edgetpu_v2_model_blocks.py 857 official/projects/edgetpu/vision/serving/__init__.py 1 official/projects/edgetpu/vision/serving/export_tflite.py 127 official/projects/edgetpu/vision/serving/export_util.py 131 official/projects/edgetpu/vision/serving/tflite_imagenet_evaluator.py 54 official/projects/edgetpu/vision/serving/tflite_imagenet_evaluator_run.py 39 official/projects/edgetpu/vision/tasks/__init__.py 1 official/projects/edgetpu/vision/tasks/image_classification.py 229 official/projects/edgetpu/vision/tasks/semantic_segmentation.py 226 official/projects/edgetpu/vision/train.py 43 official/projects/mobilebert/__init__.py 1 official/projects/mobilebert/distillation.py 409 official/projects/mobilebert/experiments/en_uncased_student.yaml 22 official/projects/mobilebert/experiments/en_uncased_teacher.yaml 22 official/projects/mobilebert/experiments/mobilebert_distillation_en_uncased.yaml 79 official/projects/mobilebert/export_tfhub.py 50 official/projects/mobilebert/model_utils.py 99 official/projects/mobilebert/run_distillation.py 91 official/projects/mobilebert/tf2_model_checkpoint_converter.py 174 official/projects/mobilebert/utils.py 10 official/projects/movinet/__init__.py 1 official/projects/movinet/configs/__init__.py 1 official/projects/movinet/configs/movinet.py 62 official/projects/movinet/configs/yaml/movinet_a0_k600_8x8.yaml 69 official/projects/movinet/configs/yaml/movinet_a0_k600_cpu_local.yaml 56 official/projects/movinet/configs/yaml/movinet_a0_stream_k600_8x8.yaml 73 official/projects/movinet/configs/yaml/movinet_a1_k600_8x8.yaml 69 official/projects/movinet/configs/yaml/movinet_a1_stream_k600_8x8.yaml 73 official/projects/movinet/configs/yaml/movinet_a2_k600_8x8.yaml 69 official/projects/movinet/configs/yaml/movinet_a2_stream_k600_8x8.yaml 73 official/projects/movinet/configs/yaml/movinet_a3_k600_8x8.yaml 69 official/projects/movinet/configs/yaml/movinet_a3_stream_k600_8x8.yaml 74 official/projects/movinet/configs/yaml/movinet_a4_k600_8x8.yaml 69 official/projects/movinet/configs/yaml/movinet_a4_stream_k600_8x8.yaml 75 official/projects/movinet/configs/yaml/movinet_a5_k600_8x8.yaml 69 official/projects/movinet/configs/yaml/movinet_a5_stream_k600_8x8.yaml 75 official/projects/movinet/configs/yaml/movinet_t0_k600_8x8.yaml 68 official/projects/movinet/configs/yaml/movinet_t0_stream_k600_8x8.yaml 73 official/projects/movinet/modeling/__init__.py 1 official/projects/movinet/modeling/movinet.py 568 official/projects/movinet/modeling/movinet_layers.py 1014 official/projects/movinet/modeling/movinet_model.py 161 official/projects/movinet/tools/convert_3d_2plus1d.py 76 official/projects/movinet/tools/export_saved_model.py 127 official/projects/movinet/train.py 43 official/projects/nhnet/__init__.py 1 official/projects/nhnet/configs.py 63 official/projects/nhnet/decoder.py 269 official/projects/nhnet/evaluation.py 97 official/projects/nhnet/input_pipeline.py 162 official/projects/nhnet/models.py 436 official/projects/nhnet/optimizer.py 30 official/projects/nhnet/raw_data_process.py 60 official/projects/nhnet/raw_data_processor.py 169 official/projects/nhnet/trainer.py 177 official/projects/nhnet/utils.py 49 official/projects/roformer/__init__.py 1 official/projects/roformer/experiments/roformer_base.yaml 61 official/projects/roformer/roformer.py 25 official/projects/roformer/roformer_attention.py 70 official/projects/roformer/roformer_encoder.py 167 official/projects/roformer/roformer_encoder_block.py 223 official/projects/roformer/roformer_experiments.py 96 official/projects/roformer/train.py 37 official/projects/teams/__init__.py 1 official/projects/teams/experiments/base/glue_mnli.yaml 39 official/projects/teams/experiments/base/squad_v1.yaml 47 official/projects/teams/experiments/base/squad_v2.yaml 48 official/projects/teams/experiments/base/wiki_books_pretrain.yaml 75 official/projects/teams/experiments/small/glue_mnli.yaml 38 official/projects/teams/experiments/small/squad_v1.yaml 46 official/projects/teams/experiments/small/squad_v2.yaml 47 official/projects/teams/experiments/small/wiki_books_pretrain.yaml 75 official/projects/teams/experiments/teams_en_uncased_base.yaml 17 official/projects/teams/experiments/teams_en_uncased_small.yaml 17 official/projects/teams/teams.py 61 official/projects/teams/teams_experiments.py 79 official/projects/teams/teams_pretrainer.py 258 official/projects/teams/teams_task.py 175 official/projects/text_classification_example/classification_data_loader.py 45 official/projects/text_classification_example/classification_example.py 127 official/projects/text_classification_example/experiments/classification_ft_cola.yaml 41 official/projects/text_classification_example/experiments/local_example.yaml 40 official/projects/text_classification_example/train.py 36 official/projects/triviaqa/__init__.py 1 official/projects/triviaqa/dataset.py 333 official/projects/triviaqa/download_and_prepare.py 44 official/projects/triviaqa/evaluate.py 24 official/projects/triviaqa/evaluation.py 119 official/projects/triviaqa/inputs.py 308 official/projects/triviaqa/modeling.py 78 official/projects/triviaqa/predict.py 139 official/projects/triviaqa/prediction.py 40 official/projects/triviaqa/preprocess.py 427 official/projects/triviaqa/sentencepiece_pb2.py 279 official/projects/triviaqa/train.py 314 official/projects/vit/configs/__init__.py 1 official/projects/vit/configs/backbones.py 24 official/projects/vit/configs/image_classification.py 235 official/projects/vit/modeling/nn_blocks.py 72 official/projects/vit/modeling/vit.py 221 official/projects/vit/train.py 8 official/projects/volumetric_models/configs/backbones.py 14 official/projects/volumetric_models/configs/decoders.py 18 official/projects/volumetric_models/configs/semantic_segmentation_3d.py 125 official/projects/volumetric_models/dataloaders/segmentation_input_3d.py 56 official/projects/volumetric_models/evaluation/segmentation_metrics.py 63 official/projects/volumetric_models/losses/segmentation_losses.py 51 official/projects/volumetric_models/modeling/backbones/__init__.py 1 official/projects/volumetric_models/modeling/backbones/unet_3d.py 106 official/projects/volumetric_models/modeling/decoders/__init__.py 1 official/projects/volumetric_models/modeling/decoders/factory.py 26 official/projects/volumetric_models/modeling/decoders/unet_3d_decoder.py 111 official/projects/volumetric_models/modeling/factory.py 36 official/projects/volumetric_models/modeling/heads/segmentation_heads_3d.py 104 official/projects/volumetric_models/modeling/nn_blocks_3d.py 357 official/projects/volumetric_models/registry_imports.py 5 official/projects/volumetric_models/serving/export_saved_model.py 74 official/projects/volumetric_models/serving/semantic_segmentation_3d.py 22 official/projects/volumetric_models/tasks/semantic_segmentation_3d.py 204 official/projects/volumetric_models/train.py 10 official/projects/yt8m/__init__.py 1 official/projects/yt8m/configs/__init__.py 1 official/projects/yt8m/configs/yt8m.py 148 official/projects/yt8m/dataloaders/utils.py 111 official/projects/yt8m/dataloaders/yt8m_input.py 233 official/projects/yt8m/eval_utils/average_precision_calculator.py 100 official/projects/yt8m/eval_utils/eval_util.py 113 official/projects/yt8m/eval_utils/mean_average_precision_calculator.py 31 official/projects/yt8m/experiments/yt8m.yaml 55 official/projects/yt8m/modeling/__init__.py 1 official/projects/yt8m/modeling/yt8m_agg_models.py 57 official/projects/yt8m/modeling/yt8m_model.py 129 official/projects/yt8m/modeling/yt8m_model_utils.py 100 official/projects/yt8m/tasks/__init__.py 1 official/projects/yt8m/tasks/yt8m_task.py 155 official/projects/yt8m/train.py 8 official/recommendation/__init__.py 1 official/recommendation/constants.py 30 official/recommendation/create_ncf_data.py 74 official/recommendation/data_pipeline.py 620 official/recommendation/data_preprocessing.py 130 official/recommendation/movielens.py 165 official/recommendation/ncf_common.py 267 official/recommendation/ncf_input_pipeline.py 84 official/recommendation/ncf_keras_main.py 361 official/recommendation/neumf_model.py 211 official/recommendation/popen_helper.py 29 official/recommendation/ranking/__init__.py 1 official/recommendation/ranking/common.py 73 official/recommendation/ranking/configs/__init__.py 1 official/recommendation/ranking/configs/config.py 188 official/recommendation/ranking/configs/yaml/dcn_v2_criteo_tpu.yaml 35 official/recommendation/ranking/configs/yaml/dlrm_criteo_tpu.yaml 35 official/recommendation/ranking/data/__init__.py 1 official/recommendation/ranking/data/data_pipeline.py 118 official/recommendation/ranking/preprocessing/criteo_preprocess.py 183 official/recommendation/ranking/preprocessing/setup.py 9 official/recommendation/ranking/preprocessing/shard_rebalancer.py 72 official/recommendation/ranking/task.py 123 official/recommendation/ranking/train.py 135 official/recommendation/stat_utils.py 32 official/utils/__init__.py 1 official/utils/flags/__init__.py 1 official/utils/flags/_base.py 121 official/utils/flags/_benchmark.py 75 official/utils/flags/_conventions.py 19 official/utils/flags/_device.py 46 official/utils/flags/_distribution.py 21 official/utils/flags/_misc.py 18 official/utils/flags/_performance.py 209 official/utils/flags/core.py 57 official/utils/hyperparams_flags.py 78 official/utils/misc/__init__.py 1 official/utils/misc/keras_utils.py 122 official/utils/misc/model_helpers.py 35 official/utils/testing/__init__.py 1 official/utils/testing/integration.py 29 official/utils/testing/mock_task.py 64 official/vision/__init__.py 2 official/vision/beta/__init__.py 2 official/vision/beta/configs/__init__.py 5 official/vision/beta/configs/backbones.py 68 official/vision/beta/configs/backbones_3d.py 65 official/vision/beta/configs/common.py 74 official/vision/beta/configs/decoders.py 33 official/vision/beta/configs/experiments/image_classification/imagenet_mobilenetv2_gpu.yaml 48 official/vision/beta/configs/experiments/image_classification/imagenet_mobilenetv2_tpu.yaml 47 official/vision/beta/configs/experiments/image_classification/imagenet_mobilenetv3large_tpu.yaml 50 official/vision/beta/configs/experiments/image_classification/imagenet_mobilenetv3small_tpu.yaml 62 official/vision/beta/configs/experiments/image_classification/imagenet_resnet101_deeplab_tpu.yaml 56 official/vision/beta/configs/experiments/image_classification/imagenet_resnet101_tpu.yaml 49 official/vision/beta/configs/experiments/image_classification/imagenet_resnet152_tpu.yaml 49 official/vision/beta/configs/experiments/image_classification/imagenet_resnet50_deeplab_tpu.yaml 50 official/vision/beta/configs/experiments/image_classification/imagenet_resnet50_gpu.yaml 48 official/vision/beta/configs/experiments/image_classification/imagenet_resnet50_tfds_tpu.yaml 55 official/vision/beta/configs/experiments/image_classification/imagenet_resnet50_tpu.yaml 47 official/vision/beta/configs/experiments/image_classification/imagenet_resnetrs101_i160.yaml 63 official/vision/beta/configs/experiments/image_classification/imagenet_resnetrs101_i192.yaml 63 official/vision/beta/configs/experiments/image_classification/imagenet_resnetrs152_i192.yaml 63 official/vision/beta/configs/experiments/image_classification/imagenet_resnetrs152_i224.yaml 63 official/vision/beta/configs/experiments/image_classification/imagenet_resnetrs152_i256.yaml 63 official/vision/beta/configs/experiments/image_classification/imagenet_resnetrs200_i256.yaml 63 official/vision/beta/configs/experiments/image_classification/imagenet_resnetrs270_i256.yaml 63 official/vision/beta/configs/experiments/image_classification/imagenet_resnetrs350_i256.yaml 63 official/vision/beta/configs/experiments/image_classification/imagenet_resnetrs350_i320.yaml 63 official/vision/beta/configs/experiments/image_classification/imagenet_resnetrs420_i320.yaml 63 official/vision/beta/configs/experiments/image_classification/imagenet_resnetrs50_i160.yaml 63 official/vision/beta/configs/experiments/maskrcnn/coco_spinenet143_cascadercnn_tpu.yaml 56 official/vision/beta/configs/experiments/maskrcnn/coco_spinenet143_mrcnn_tpu.yaml 46 official/vision/beta/configs/experiments/maskrcnn/coco_spinenet49_cascadercnn_tpu.yaml 56 official/vision/beta/configs/experiments/maskrcnn/coco_spinenet49_mrcnn_tpu.yaml 46 official/vision/beta/configs/experiments/maskrcnn/coco_spinenet96_cascadercnn_tpu.yaml 56 official/vision/beta/configs/experiments/maskrcnn/coco_spinenet96_mrcnn_tpu.yaml 46 official/vision/beta/configs/experiments/maskrcnn/r50fpn_640_coco_scratch_tpu4x4.yaml 35 official/vision/beta/configs/experiments/retinanet/coco_mobiledetcpu_tpu.yaml 60 official/vision/beta/configs/experiments/retinanet/coco_mobilenetv2_tpu.yaml 60 official/vision/beta/configs/experiments/retinanet/coco_spinenet143_tpu.yaml 57 official/vision/beta/configs/experiments/retinanet/coco_spinenet190_tpu.yaml 57 official/vision/beta/configs/experiments/retinanet/coco_spinenet49_mobile_tpu.yaml 59 official/vision/beta/configs/experiments/retinanet/coco_spinenet49_tpu.yaml 57 official/vision/beta/configs/experiments/retinanet/coco_spinenet49s_mobile_tpu.yaml 59 official/vision/beta/configs/experiments/retinanet/coco_spinenet49xs_mobile_tpu.yaml 59 official/vision/beta/configs/experiments/retinanet/coco_spinenet96_tpu.yaml 57 official/vision/beta/configs/experiments/retinanet/resnet50fpn_coco_tfds_tpu.yaml 34 official/vision/beta/configs/experiments/retinanet/resnet50fpn_coco_tpu4x4_benchmark.yaml 6 official/vision/beta/configs/experiments/semantic_segmentation/deeplabv3plus_resnet101_cityscapes_tfds_tpu.yaml 77 official/vision/beta/configs/experiments/video_classification/k400_3d-resnet50_tpu.yaml 84 official/vision/beta/configs/experiments/video_classification/k400_resnet3drs_50_tpu.yaml 95 official/vision/beta/configs/experiments/video_classification/k400_slowonly16x4_tpu.yaml 84 official/vision/beta/configs/experiments/video_classification/k400_slowonly8x8_tpu.yaml 84 official/vision/beta/configs/experiments/video_classification/k600_3d-resnet50_tpu.yaml 84 official/vision/beta/configs/experiments/video_classification/k600_3d-resnet50g_tpu.yaml 107 official/vision/beta/configs/experiments/video_classification/k600_slowonly8x8_tpu.yaml 84 official/vision/beta/configs/image_classification.py 344 official/vision/beta/configs/maskrcnn.py 447 official/vision/beta/configs/retinanet.py 346 official/vision/beta/configs/semantic_segmentation.py 623 official/vision/beta/configs/video_classification.py 308 official/vision/beta/data/__init__.py 1 official/vision/beta/data/create_coco_tf_record.py 355 official/vision/beta/data/process_coco_few_shot_json_files.py 84 official/vision/beta/data/tfrecord_lib.py 95 official/vision/beta/dataloaders/__init__.py 1 official/vision/beta/dataloaders/classification_input.py 176 official/vision/beta/dataloaders/decoder.py 6 official/vision/beta/dataloaders/input_reader.py 88 official/vision/beta/dataloaders/input_reader_factory.py 14 official/vision/beta/dataloaders/maskrcnn_input.py 179 official/vision/beta/dataloaders/parser.py 19 official/vision/beta/dataloaders/retinanet_input.py 198 official/vision/beta/dataloaders/segmentation_input.py 124 official/vision/beta/dataloaders/tf_example_decoder.py 116 official/vision/beta/dataloaders/tf_example_label_map_decoder.py 39 official/vision/beta/dataloaders/tfds_classification_decoders.py 16 official/vision/beta/dataloaders/tfds_detection_decoders.py 21 official/vision/beta/dataloaders/tfds_factory.py 23 official/vision/beta/dataloaders/tfds_segmentation_decoders.py 61 official/vision/beta/dataloaders/tfexample_utils.py 164 official/vision/beta/dataloaders/utils.py 27 official/vision/beta/dataloaders/video_input.py 233 official/vision/beta/evaluation/__init__.py 1 official/vision/beta/evaluation/coco_evaluator.py 202 official/vision/beta/evaluation/coco_utils.py 234 official/vision/beta/evaluation/iou.py 44 official/vision/beta/evaluation/panoptic_quality.py 134 official/vision/beta/evaluation/panoptic_quality_evaluator.py 80 official/vision/beta/evaluation/segmentation_metrics.py 115 official/vision/beta/evaluation/wod_detection_evaluator.py 101 official/vision/beta/losses/__init__.py 1 official/vision/beta/losses/focal_loss.py 31 official/vision/beta/losses/loss_utils.py 14 official/vision/beta/losses/maskrcnn_losses.py 130 official/vision/beta/losses/retinanet_losses.py 81 official/vision/beta/losses/segmentation_losses.py 89 official/vision/beta/modeling/__init__.py 4 official/vision/beta/modeling/backbones/__init__.py 9 official/vision/beta/modeling/backbones/efficientnet.py 216 official/vision/beta/modeling/backbones/factory.py 21 official/vision/beta/modeling/backbones/mobiledet.py 414 official/vision/beta/modeling/backbones/mobilenet.py 682 official/vision/beta/modeling/backbones/resnet.py 326 official/vision/beta/modeling/backbones/resnet_3d.py 343 official/vision/beta/modeling/backbones/resnet_deeplab.py 261 official/vision/beta/modeling/backbones/revnet.py 145 official/vision/beta/modeling/backbones/spinenet.py 448 official/vision/beta/modeling/backbones/spinenet_mobile.py 397 official/vision/beta/modeling/classification_model.py 71 official/vision/beta/modeling/decoders/__init__.py 3 official/vision/beta/modeling/decoders/aspp.py 107 official/vision/beta/modeling/decoders/factory.py 25 official/vision/beta/modeling/decoders/fpn.py 152 official/vision/beta/modeling/decoders/nasfpn.py 248 official/vision/beta/modeling/factory.py 326 official/vision/beta/modeling/factory_3d.py 38 official/vision/beta/modeling/heads/__init__.py 5 official/vision/beta/modeling/heads/dense_prediction_heads.py 341 official/vision/beta/modeling/heads/instance_heads.py 303 official/vision/beta/modeling/heads/segmentation_heads.py 278 official/vision/beta/modeling/layers/__init__.py 27 official/vision/beta/modeling/layers/box_sampler.py 40 official/vision/beta/modeling/layers/deeplab.py 145 official/vision/beta/modeling/layers/detection_generator.py 468 official/vision/beta/modeling/layers/mask_sampler.py 67 official/vision/beta/modeling/layers/nn_blocks.py 1071 official/vision/beta/modeling/layers/nn_blocks_3d.py 192 official/vision/beta/modeling/layers/nn_layers.py 700 official/vision/beta/modeling/layers/roi_aligner.py 26 official/vision/beta/modeling/layers/roi_generator.py 154 official/vision/beta/modeling/layers/roi_sampler.py 85 official/vision/beta/modeling/maskrcnn_model.py 291 official/vision/beta/modeling/retinanet_model.py 120 official/vision/beta/modeling/segmentation_model.py 47 official/vision/beta/modeling/video_classification_model.py 78 official/vision/beta/ops/__init__.py 1 official/vision/beta/ops/anchor.py 197 official/vision/beta/ops/anchor_generator.py 86 official/vision/beta/ops/augment.py 1238 official/vision/beta/ops/box_matcher.py 75 official/vision/beta/ops/box_ops.py 353 official/vision/beta/ops/iou_similarity.py 73 official/vision/beta/ops/mask_ops.py 102 official/vision/beta/ops/nms.py 97 official/vision/beta/ops/preprocess_ops.py 447 official/vision/beta/ops/preprocess_ops_3d.py 178 official/vision/beta/ops/sampling_ops.py 166 official/vision/beta/ops/spatial_transform_ops.py 281 official/vision/beta/ops/target_gather.py 57 official/vision/beta/projects/__init__.py 1 official/vision/beta/projects/centernet/common/registry_imports.py 5 official/vision/beta/projects/centernet/configs/__init__.py 1 official/vision/beta/projects/centernet/configs/backbones.py 13 official/vision/beta/projects/centernet/configs/centernet.py 168 official/vision/beta/projects/centernet/configs/experiments/coco-centernet-hourglass-gpu.yaml 84 official/vision/beta/projects/centernet/configs/experiments/coco-centernet-hourglass-tpu.yaml 82 official/vision/beta/projects/centernet/dataloaders/centernet_input.py 217 official/vision/beta/projects/centernet/losses/centernet_losses.py 29 official/vision/beta/projects/centernet/modeling/backbones/hourglass.py 199 official/vision/beta/projects/centernet/modeling/centernet_model.py 47 official/vision/beta/projects/centernet/modeling/heads/centernet_head.py 49 official/vision/beta/projects/centernet/modeling/layers/cn_nn_blocks.py 195 official/vision/beta/projects/centernet/modeling/layers/detection_generator.py 162 official/vision/beta/projects/centernet/ops/__init__.py 1 official/vision/beta/projects/centernet/ops/box_list.py 63 official/vision/beta/projects/centernet/ops/box_list_ops.py 134 official/vision/beta/projects/centernet/ops/loss_ops.py 57 official/vision/beta/projects/centernet/ops/nms_ops.py 75 official/vision/beta/projects/centernet/ops/preprocess_ops.py 235 official/vision/beta/projects/centernet/ops/target_assigner.py 144 official/vision/beta/projects/centernet/tasks/centernet.py 300 official/vision/beta/projects/centernet/train.py 36 official/vision/beta/projects/centernet/utils/checkpoints/__init__.py 1 official/vision/beta/projects/centernet/utils/checkpoints/config_classes.py 194 official/vision/beta/projects/centernet/utils/checkpoints/config_data.py 71 official/vision/beta/projects/centernet/utils/checkpoints/load_weights.py 76 official/vision/beta/projects/centernet/utils/checkpoints/read_checkpoints.py 41 official/vision/beta/projects/centernet/utils/tf2_centernet_checkpoint_converter.py 96 official/vision/beta/projects/deepmac_maskrcnn/__init__.py 1 official/vision/beta/projects/deepmac_maskrcnn/common/__init__.py 1 official/vision/beta/projects/deepmac_maskrcnn/common/registry_imports.py 1 official/vision/beta/projects/deepmac_maskrcnn/configs/__init__.py 1 official/vision/beta/projects/deepmac_maskrcnn/configs/deep_mask_head_rcnn.py 163 official/vision/beta/projects/deepmac_maskrcnn/configs/experiments/deep_mask_head_rcnn_nonvoc_spinenet143_hg52.yaml 60 official/vision/beta/projects/deepmac_maskrcnn/configs/experiments/deep_mask_head_rcnn_voc_r101_hg52.yaml 28 official/vision/beta/projects/deepmac_maskrcnn/configs/experiments/deep_mask_head_rcnn_voc_r50.yaml 24 official/vision/beta/projects/deepmac_maskrcnn/configs/experiments/deep_mask_head_rcnn_voc_r50_hg52.yaml 28 official/vision/beta/projects/deepmac_maskrcnn/configs/experiments/deep_mask_head_rcnn_voc_spinenet143_hg52.yaml 55 official/vision/beta/projects/deepmac_maskrcnn/modeling/__init__.py 1 official/vision/beta/projects/deepmac_maskrcnn/modeling/heads/__init__.py 1 official/vision/beta/projects/deepmac_maskrcnn/modeling/heads/hourglass_network.py 339 official/vision/beta/projects/deepmac_maskrcnn/modeling/heads/instance_heads.py 210 official/vision/beta/projects/deepmac_maskrcnn/modeling/maskrcnn_model.py 144 official/vision/beta/projects/deepmac_maskrcnn/serving/__init__.py 1 official/vision/beta/projects/deepmac_maskrcnn/serving/detection.py 59 official/vision/beta/projects/deepmac_maskrcnn/serving/export_saved_model.py 61 official/vision/beta/projects/deepmac_maskrcnn/tasks/__init__.py 1 official/vision/beta/projects/deepmac_maskrcnn/tasks/deep_mask_head_rcnn.py 146 official/vision/beta/projects/deepmac_maskrcnn/train.py 38 official/vision/beta/projects/example/example_config.py 83 official/vision/beta/projects/example/example_config_local.yaml 32 official/vision/beta/projects/example/example_config_tpu.yaml 35 official/vision/beta/projects/example/example_input.py 45 official/vision/beta/projects/example/example_model.py 36 official/vision/beta/projects/example/example_task.py 93 official/vision/beta/projects/example/registry_imports.py 5 official/vision/beta/projects/example/train.py 7 official/vision/beta/projects/panoptic_maskrcnn/__init__.py 1 official/vision/beta/projects/panoptic_maskrcnn/configs/__init__.py 1 official/vision/beta/projects/panoptic_maskrcnn/configs/experiments/r50fpn_1x_coco.yaml 23 official/vision/beta/projects/panoptic_maskrcnn/configs/experiments/r50fpn_3x_coco.yaml 23 official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py 179 official/vision/beta/projects/panoptic_maskrcnn/dataloaders/panoptic_maskrcnn_input.py 182 official/vision/beta/projects/panoptic_maskrcnn/modeling/factory.py 99 official/vision/beta/projects/panoptic_maskrcnn/modeling/layers/panoptic_segmentation_generator.py 184 official/vision/beta/projects/panoptic_maskrcnn/modeling/layers/paste_masks.py 112 official/vision/beta/projects/panoptic_maskrcnn/modeling/panoptic_maskrcnn_model.py 118 official/vision/beta/projects/panoptic_maskrcnn/serving/export_saved_model.py 69 official/vision/beta/projects/panoptic_maskrcnn/serving/panoptic_segmentation.py 97 official/vision/beta/projects/panoptic_maskrcnn/tasks/__init__.py 1 official/vision/beta/projects/panoptic_maskrcnn/tasks/panoptic_maskrcnn.py 340 official/vision/beta/projects/panoptic_maskrcnn/train.py 8 official/vision/beta/projects/simclr/common/registry_imports.py 5 official/vision/beta/projects/simclr/configs/experiments/cifar_simclr_pretrain.yaml 78 official/vision/beta/projects/simclr/configs/experiments/imagenet_simclr_finetune_gpu.yaml 71 official/vision/beta/projects/simclr/configs/experiments/imagenet_simclr_finetune_tpu.yaml 69 official/vision/beta/projects/simclr/configs/experiments/imagenet_simclr_multitask_tpu.yaml 134 official/vision/beta/projects/simclr/configs/experiments/imagenet_simclr_pretrain_gpu.yaml 72 official/vision/beta/projects/simclr/configs/experiments/imagenet_simclr_pretrain_tpu.yaml 70 official/vision/beta/projects/simclr/configs/multitask_config.py 48 official/vision/beta/projects/simclr/configs/simclr.py 251 official/vision/beta/projects/simclr/dataloaders/preprocess_ops.py 176 official/vision/beta/projects/simclr/dataloaders/simclr_input.py 122 official/vision/beta/projects/simclr/heads/simclr_head.py 125 official/vision/beta/projects/simclr/losses/contrastive_losses.py 61 official/vision/beta/projects/simclr/modeling/layers/nn_blocks.py 81 official/vision/beta/projects/simclr/modeling/multitask_model.py 82 official/vision/beta/projects/simclr/modeling/simclr_model.py 100 official/vision/beta/projects/simclr/multitask_train.py 40 official/vision/beta/projects/simclr/tasks/simclr.py 424 official/vision/beta/projects/simclr/train.py 36 official/vision/beta/projects/video_ssl/configs/__init__.py 1 official/vision/beta/projects/video_ssl/configs/experiments/cvrl_linear_eval_k600.yaml 91 official/vision/beta/projects/video_ssl/configs/experiments/cvrl_pretrain_k600_200ep.yaml 73 official/vision/beta/projects/video_ssl/configs/video_ssl.py 93 official/vision/beta/projects/video_ssl/dataloaders/video_ssl_input.py 174 official/vision/beta/projects/video_ssl/losses/losses.py 66 official/vision/beta/projects/video_ssl/modeling/video_ssl_model.py 115 official/vision/beta/projects/video_ssl/ops/video_ssl_preprocess_ops.py 226 official/vision/beta/projects/video_ssl/tasks/__init__.py 2 official/vision/beta/projects/video_ssl/tasks/linear_eval.py 31 official/vision/beta/projects/video_ssl/tasks/pretrain.py 104 official/vision/beta/projects/video_ssl/train.py 44 official/vision/beta/projects/yolo/common/registry_imports.py 10 official/vision/beta/projects/yolo/configs/backbones.py 16 official/vision/beta/projects/yolo/configs/darknet_classification.py 41 official/vision/beta/projects/yolo/configs/decoders.py 24 official/vision/beta/projects/yolo/configs/experiments/darknet/csp_darknet53.yaml 51 official/vision/beta/projects/yolo/configs/experiments/darknet/csp_darknet53_tfds.yaml 58 official/vision/beta/projects/yolo/configs/experiments/darknet/darknet53.yaml 50 official/vision/beta/projects/yolo/configs/experiments/darknet/darknet53_tfds.yaml 58 official/vision/beta/projects/yolo/configs/experiments/scaled-yolo/detection/yolo_csp_640_tpu.yaml 72 official/vision/beta/projects/yolo/configs/experiments/scaled-yolo/detection/yolo_l_p5_896_tpu.yaml 83 official/vision/beta/projects/yolo/configs/experiments/scaled-yolo/detection/yolo_l_p6_1280_tpu.yaml 85 official/vision/beta/projects/yolo/configs/experiments/scaled-yolo/detection/yolo_l_p7_1536_tpu.yaml 87 official/vision/beta/projects/yolo/configs/experiments/scaled-yolo/tpu/640.yaml 72 official/vision/beta/projects/yolo/configs/experiments/yolov4/detection/yolov4_512_tpu.yaml 136 official/vision/beta/projects/yolo/configs/experiments/yolov4/imagenet_pretraining/cspdarknet53_256_tpu.yaml 51 official/vision/beta/projects/yolo/configs/yolo.py 422 official/vision/beta/projects/yolo/dataloaders/__init__.py 1 official/vision/beta/projects/yolo/dataloaders/classification_input.py 47 official/vision/beta/projects/yolo/dataloaders/tf_example_decoder.py 48 official/vision/beta/projects/yolo/dataloaders/yolo_input.py 234 official/vision/beta/projects/yolo/losses/__init__.py 1 official/vision/beta/projects/yolo/losses/yolo_loss.py 392 official/vision/beta/projects/yolo/modeling/backbones/darknet.py 568 official/vision/beta/projects/yolo/modeling/decoders/__init__.py 1 official/vision/beta/projects/yolo/modeling/decoders/yolo_decoder.py 416 official/vision/beta/projects/yolo/modeling/factory.py 65 official/vision/beta/projects/yolo/modeling/heads/__init__.py 1 official/vision/beta/projects/yolo/modeling/heads/yolo_head.py 94 official/vision/beta/projects/yolo/modeling/layers/detection_generator.py 174 official/vision/beta/projects/yolo/modeling/layers/nn_blocks.py 1104 official/vision/beta/projects/yolo/modeling/yolo_model.py 66 official/vision/beta/projects/yolo/ops/__init__.py 1 official/vision/beta/projects/yolo/ops/anchor.py 313 official/vision/beta/projects/yolo/ops/box_ops.py 139 official/vision/beta/projects/yolo/ops/loss_utils.py 302 official/vision/beta/projects/yolo/ops/math_ops.py 13 official/vision/beta/projects/yolo/ops/mosaic.py 291 official/vision/beta/projects/yolo/ops/preprocessing_ops.py 485 official/vision/beta/projects/yolo/optimization/__init__.py 5 official/vision/beta/projects/yolo/optimization/configs/__init__.py 1 official/vision/beta/projects/yolo/optimization/configs/optimization_config.py 12 official/vision/beta/projects/yolo/optimization/configs/optimizer_config.py 22 official/vision/beta/projects/yolo/optimization/optimizer_factory.py 30 official/vision/beta/projects/yolo/optimization/sgd_torch.py 191 official/vision/beta/projects/yolo/tasks/image_classification.py 40 official/vision/beta/projects/yolo/tasks/task_utils.py 29 official/vision/beta/projects/yolo/tasks/yolo.py 261 official/vision/beta/projects/yolo/train.py 9 official/vision/beta/serving/__init__.py 1 official/vision/beta/serving/detection.py 124 official/vision/beta/serving/export_base.py 113 official/vision/beta/serving/export_base_v2.py 30 official/vision/beta/serving/export_module_factory.py 57 official/vision/beta/serving/export_saved_model.py 59 official/vision/beta/serving/export_saved_model_lib.py 109 official/vision/beta/serving/export_saved_model_lib_v2.py 47 official/vision/beta/serving/export_tfhub.py 73 official/vision/beta/serving/export_tflite.py 64 official/vision/beta/serving/export_tflite_lib.py 48 official/vision/beta/serving/export_utils.py 65 official/vision/beta/serving/image_classification.py 39 official/vision/beta/serving/semantic_segmentation.py 46 official/vision/beta/serving/video_classification.py 138 official/vision/beta/tasks/__init__.py 5 official/vision/beta/tasks/image_classification.py 210 official/vision/beta/tasks/maskrcnn.py 329 official/vision/beta/tasks/retinanet.py 244 official/vision/beta/tasks/semantic_segmentation.py 208 official/vision/beta/tasks/video_classification.py 236 official/vision/beta/train.py 37 official/vision/beta/train_spatial_partitioning.py 97 official/vision/detection/__init__.py 1 official/vision/image_classification/__init__.py 2 official/vision/utils/__init__.py 1 official/vision/utils/object_detection/__init__.py 1 official/vision/utils/object_detection/argmax_matcher.py 86 official/vision/utils/object_detection/balanced_positive_negative_sampler.py 131 official/vision/utils/object_detection/box_coder.py 38 official/vision/utils/object_detection/box_list.py 61 official/vision/utils/object_detection/box_list_ops.py 465 official/vision/utils/object_detection/faster_rcnn_box_coder.py 48 official/vision/utils/object_detection/matcher.py 55 official/vision/utils/object_detection/minibatch_sampler.py 22 official/vision/utils/object_detection/ops.py 22 official/vision/utils/object_detection/preprocessor.py 208 official/vision/utils/object_detection/region_similarity_calculator.py 50 official/vision/utils/object_detection/shape_utils.py 42 official/vision/utils/object_detection/target_assigner.py 220 official/vision/utils/object_detection/visualization_utils.py 429 orbit/__init__.py 10 orbit/actions/__init__.py 5 orbit/actions/conditional_action.py 21 orbit/actions/export_saved_model.py 52 orbit/actions/new_best_metric.py 73 orbit/controller.py 216 orbit/runner.py 13 orbit/standard_runner.py 132 orbit/utils/__init__.py 9 orbit/utils/common.py 30 orbit/utils/epoch_helper.py 31 orbit/utils/loop_fns.py 61 orbit/utils/summary_manager.py 36 orbit/utils/tpu_summaries.py 30 research/adversarial_text/__init__.py 1 research/adversarial_text/adversarial_losses.py 129 research/adversarial_text/data/__init__.py 1 research/adversarial_text/data/data_utils.py 190 research/adversarial_text/data/document_generators.py 200 research/adversarial_text/evaluate.py 76 research/adversarial_text/gen_data.py 128 research/adversarial_text/gen_vocab.py 46 research/adversarial_text/graphs.py 409 research/adversarial_text/inputs.py 216 research/adversarial_text/layers.py 248 research/adversarial_text/pretrain.py 15 research/adversarial_text/train_classifier.py 23 research/adversarial_text/train_utils.py 84 research/attention_ocr/python/common_flags.py 95 research/attention_ocr/python/data_provider.py 97 research/attention_ocr/python/datasets/__init__.py 3 research/attention_ocr/python/datasets/fsns.py 121 research/attention_ocr/python/datasets/unittest_utils.py 24 research/attention_ocr/python/demo_inference.py 61 research/attention_ocr/python/eval.py 45 research/attention_ocr/python/inception_preprocessing.py 129 research/attention_ocr/python/metrics.py 39 research/attention_ocr/python/model.py 366 research/attention_ocr/python/model_export.py 119 research/attention_ocr/python/model_export_lib.py 52 research/attention_ocr/python/sequence_layers.py 169 research/attention_ocr/python/train.py 141 research/attention_ocr/python/utils.py 32 research/audioset/vggish/mel_features.py 69 research/audioset/vggish/vggish_export_tfhub.py 83 research/audioset/vggish/vggish_inference_demo.py 77 research/audioset/vggish/vggish_input.py 41 research/audioset/vggish/vggish_params.py 24 research/audioset/vggish/vggish_postprocess.py 28 research/audioset/vggish/vggish_slim.py 42 research/audioset/vggish/vggish_train_demo.py 79 research/audioset/yamnet/export.py 139 research/audioset/yamnet/features.py 101 research/audioset/yamnet/inference.py 31 research/audioset/yamnet/params.py 25 research/audioset/yamnet/yamnet.py 91 research/autoaugment/augmentation_transforms.py 202 research/autoaugment/custom_ops.py 120 research/autoaugment/data_utils.py 134 research/autoaugment/helper_utils.py 71 research/autoaugment/policies.py 122 research/autoaugment/shake_drop.py 116 research/autoaugment/shake_shake.py 86 research/autoaugment/train_cifar.py 299 research/autoaugment/wrn.py 83 research/cognitive_planning/__init__.py 1 research/cognitive_planning/embedders.py 327 research/cognitive_planning/envs/__init__.py 1 research/cognitive_planning/envs/active_vision_dataset_env.py 656 research/cognitive_planning/envs/task_env.py 79 research/cognitive_planning/envs/util.py 15 research/cognitive_planning/label_map_util.py 70 research/cognitive_planning/policies.py 232 research/cognitive_planning/preprocessing/__init__.py 1 research/cognitive_planning/preprocessing/cifarnet_preprocessing.py 47 research/cognitive_planning/preprocessing/inception_preprocessing.py 129 research/cognitive_planning/preprocessing/lenet_preprocessing.py 12 research/cognitive_planning/preprocessing/preprocessing_factory.py 42 research/cognitive_planning/preprocessing/vgg_preprocessing.py 144 research/cognitive_planning/standard_fields.py 83 research/cognitive_planning/string_int_label_map_pb2.py 100 research/cognitive_planning/tasks.py 773 research/cognitive_planning/train_supervised_active_vision.py 360 research/cognitive_planning/visualization_utils.py 427 research/cognitive_planning/viz_active_vision_dataset_main.py 296 research/cvt_text/__init__.py 1 research/cvt_text/base/__init__.py 1 research/cvt_text/base/configure.py 80 research/cvt_text/base/embeddings.py 111 research/cvt_text/base/utils.py 35 research/cvt_text/corpus_processing/__init__.py 1 research/cvt_text/corpus_processing/example.py 27 research/cvt_text/corpus_processing/minibatching.py 101 research/cvt_text/corpus_processing/scorer.py 26 research/cvt_text/corpus_processing/unlabeled_data.py 56 research/cvt_text/cvt.py 42 research/cvt_text/model/__init__.py 1 research/cvt_text/model/encoder.py 84 research/cvt_text/model/model_helpers.py 26 research/cvt_text/model/multitask_model.py 96 research/cvt_text/model/shared_inputs.py 27 research/cvt_text/model/task_module.py 18 research/cvt_text/preprocessing.py 54 research/cvt_text/task_specific/__init__.py 1 research/cvt_text/task_specific/task_definitions.py 57 research/cvt_text/task_specific/word_level/__init__.py 1 research/cvt_text/task_specific/word_level/depparse_module.py 88 research/cvt_text/task_specific/word_level/depparse_scorer.py 24 research/cvt_text/task_specific/word_level/tagging_module.py 50 research/cvt_text/task_specific/word_level/tagging_scorers.py 54 research/cvt_text/task_specific/word_level/tagging_utils.py 32 research/cvt_text/task_specific/word_level/word_level_data.py 126 research/cvt_text/task_specific/word_level/word_level_scorer.py 24 research/cvt_text/training/__init__.py 1 research/cvt_text/training/trainer.py 102 research/cvt_text/training/training_progress.py 52 research/deep_speech/__init__.py 1 research/deep_speech/data/__init__.py 1 research/deep_speech/data/dataset.py 127 research/deep_speech/data/download.py 118 research/deep_speech/data/featurizer.py 63 research/deep_speech/decoder.py 29 research/deep_speech/deep_speech.py 243 research/deep_speech/deep_speech_model.py 69 research/deeplab/__init__.py 1 research/deeplab/common.py 202 research/deeplab/convert_to_tflite.py 70 research/deeplab/core/__init__.py 1 research/deeplab/core/conv2d_ws.py 162 research/deeplab/core/dense_prediction_cell.py 156 research/deeplab/core/feature_extractor.py 477 research/deeplab/core/nas_cell.py 157 research/deeplab/core/nas_genotypes.py 20 research/deeplab/core/nas_network.py 270 research/deeplab/core/preprocess_utils.py 265 research/deeplab/core/resnet_v1_beta.py 396 research/deeplab/core/utils.py 83 research/deeplab/core/xception.py 596 research/deeplab/datasets/__init__.py 1 research/deeplab/datasets/build_ade20k_data.py 74 research/deeplab/datasets/build_cityscapes_data.py 88 research/deeplab/datasets/build_data.py 57 research/deeplab/datasets/build_voc2012_data.py 67 research/deeplab/datasets/data_generator.py 184 research/deeplab/datasets/remove_gt_colormap.py 33 research/deeplab/deprecated/__init__.py 1 research/deeplab/deprecated/segmentation_dataset.py 104 research/deeplab/eval.py 155 research/deeplab/evaluation/__init__.py 1 research/deeplab/evaluation/base_metric.py 44 research/deeplab/evaluation/eval_coco_format.py 216 research/deeplab/evaluation/panoptic_quality.py 142 research/deeplab/evaluation/parsing_covering.py 121 research/deeplab/evaluation/streaming_metrics.py 133 research/deeplab/export_model.py 122 research/deeplab/input_preprocess.py 65 research/deeplab/model.py 571 research/deeplab/train.py 295 research/deeplab/utils/__init__.py 1 research/deeplab/utils/get_dataset_colormap.py 300 research/deeplab/utils/save_annotation.py 27 research/deeplab/utils/train_utils.py 190 research/deeplab/vis.py 202 research/delf/delf/__init__.py 24 research/delf/delf/protos/__init__.py 1 research/delf/delf/protos/aggregation_config.proto 55 research/delf/delf/protos/box.proto 18 research/delf/delf/protos/datum.proto 58 research/delf/delf/protos/delf_config.proto 107 research/delf/delf/protos/feature.proto 17 research/delf/delf/python/__init__.py 1 research/delf/delf/python/box_io.py 47 research/delf/delf/python/datasets/generic_dataset.py 24 research/delf/delf/python/datasets/google_landmarks_dataset/__init__.py 4 research/delf/delf/python/datasets/google_landmarks_dataset/compute_recognition_metrics.py 59 research/delf/delf/python/datasets/google_landmarks_dataset/compute_retrieval_metrics.py 66 research/delf/delf/python/datasets/google_landmarks_dataset/dataset_file_io.py 76 research/delf/delf/python/datasets/google_landmarks_dataset/googlelandmarks.py 82 research/delf/delf/python/datasets/google_landmarks_dataset/metrics.py 96 research/delf/delf/python/datasets/revisited_op/__init__.py 4 research/delf/delf/python/datasets/revisited_op/dataset.py 219 research/delf/delf/python/datasets/sfm120k/__init__.py 4 research/delf/delf/python/datasets/sfm120k/dataset_download.py 62 research/delf/delf/python/datasets/sfm120k/sfm120k.py 57 research/delf/delf/python/datasets/tuples_dataset.py 154 research/delf/delf/python/datasets/utils.py 18 research/delf/delf/python/datum_io.py 62 research/delf/delf/python/delg/extract_features.py 108 research/delf/delf/python/delg/measure_latency.py 70 research/delf/delf/python/delg/perform_retrieval.py 158 research/delf/delf/python/detect_to_retrieve/__init__.py 5 research/delf/delf/python/detect_to_retrieve/aggregation_extraction.py 108 research/delf/delf/python/detect_to_retrieve/boxes_and_features_extraction.py 117 research/delf/delf/python/detect_to_retrieve/cluster_delf_features.py 128 research/delf/delf/python/detect_to_retrieve/extract_aggregation.py 68 research/delf/delf/python/detect_to_retrieve/extract_index_boxes_and_features.py 84 research/delf/delf/python/detect_to_retrieve/extract_query_features.py 83 research/delf/delf/python/detect_to_retrieve/image_reranking.py 160 research/delf/delf/python/detect_to_retrieve/perform_retrieval.py 206 research/delf/delf/python/feature_aggregation_extractor.py 235 research/delf/delf/python/feature_aggregation_similarity.py 129 research/delf/delf/python/feature_extractor.py 58 research/delf/delf/python/feature_io.py 75 research/delf/delf/python/normalization_layers/__init__.py 1 research/delf/delf/python/normalization_layers/normalization.py 7 research/delf/delf/python/pooling_layers/__init__.py 1 research/delf/delf/python/pooling_layers/pooling.py 58 research/delf/delf/python/training/__init__.py 4 research/delf/delf/python/training/build_image_dataset.py 238 research/delf/delf/python/training/global_features/__init__.py 3 research/delf/delf/python/training/global_features/train.py 255 research/delf/delf/python/training/global_features/train_utils.py 207 research/delf/delf/python/training/global_features_utils.py 100 research/delf/delf/python/training/losses/__init__.py 1 research/delf/delf/python/training/losses/ranking_losses.py 43 research/delf/delf/python/training/model/__init__.py 7 research/delf/delf/python/training/model/delf_model.py 119 research/delf/delf/python/training/model/delg_model.py 82 research/delf/delf/python/training/model/export_CNN_global.py 109 research/delf/delf/python/training/model/export_global_model.py 119 research/delf/delf/python/training/model/export_local_and_global_model.py 112 research/delf/delf/python/training/model/export_local_model.py 76 research/delf/delf/python/training/model/export_model_utils.py 187 research/delf/delf/python/training/model/global_model.py 142 research/delf/delf/python/training/model/resnet50.py 285 research/delf/delf/python/training/tensorboard_utils.py 7 research/delf/delf/python/training/train.py 373 research/delf/delf/python/utils.py 44 research/delf/delf/python/whiten.py 43 research/delf/setup.py 19 research/efficient-hrl/agent.py 481 research/efficient-hrl/agents/__init__.py 1 research/efficient-hrl/agents/circular_buffer.py 143 research/efficient-hrl/agents/ddpg_agent.py 317 research/efficient-hrl/agents/ddpg_networks.py 83 research/efficient-hrl/cond_fn.py 89 research/efficient-hrl/context/__init__.py 1 research/efficient-hrl/context/context.py 301 research/efficient-hrl/context/context_transition_functions.py 60 research/efficient-hrl/context/gin_imports.py 5 research/efficient-hrl/context/gin_utils.py 20 research/efficient-hrl/context/rewards_functions.py 368 research/efficient-hrl/context/samplers.py 254 research/efficient-hrl/environments/__init__.py 1 research/efficient-hrl/environments/ant.py 98 research/efficient-hrl/environments/ant_maze_env.py 4 research/efficient-hrl/environments/create_maze_env.py 70 research/efficient-hrl/environments/maze_env.py 419 research/efficient-hrl/environments/maze_env_utils.py 101 research/efficient-hrl/environments/point.py 61 research/efficient-hrl/environments/point_maze_env.py 4 research/efficient-hrl/eval.py 298 research/efficient-hrl/run_env.py 83 research/efficient-hrl/run_eval.py 21 research/efficient-hrl/run_train.py 20 research/efficient-hrl/scripts/local_eval.py 52 research/efficient-hrl/scripts/local_train.py 52 research/efficient-hrl/train.py 547 research/efficient-hrl/train_utils.py 104 research/efficient-hrl/utils/__init__.py 1 research/efficient-hrl/utils/eval_utils.py 95 research/efficient-hrl/utils/utils.py 170 research/lfads/distributions.py 191 research/lfads/lfads.py 1386 research/lfads/plot_lfads.py 133 research/lfads/run_lfads.py 444 research/lfads/synth_data/generate_chaotic_rnn_data.py 128 research/lfads/synth_data/generate_itb_data.py 152 research/lfads/synth_data/generate_labeled_rnn_data.py 107 research/lfads/synth_data/synthetic_data_utils.py 192 research/lfads/utils.py 169 research/lstm_object_detection/__init__.py 1 research/lstm_object_detection/builders/__init__.py 1 research/lstm_object_detection/builders/graph_rewriter_builder.py 81 research/lstm_object_detection/eval.py 71 research/lstm_object_detection/evaluator.py 208 research/lstm_object_detection/export_tflite_lstd_graph.py 44 research/lstm_object_detection/export_tflite_lstd_graph_lib.py 195 research/lstm_object_detection/export_tflite_lstd_model.py 38 research/lstm_object_detection/inputs/__init__.py 1 research/lstm_object_detection/inputs/seq_dataset_builder.py 159 research/lstm_object_detection/inputs/tf_sequence_example_decoder.py 134 research/lstm_object_detection/lstm/__init__.py 1 research/lstm_object_detection/lstm/lstm_cells.py 442 research/lstm_object_detection/lstm/rnn_decoder.py 114 research/lstm_object_detection/lstm/utils.py 140 research/lstm_object_detection/meta_architectures/__init__.py 1 research/lstm_object_detection/meta_architectures/lstm_ssd_meta_arch.py 271 research/lstm_object_detection/metrics/__init__.py 1 research/lstm_object_detection/metrics/coco_evaluation_all_frames.py 51 research/lstm_object_detection/model_builder.py 112 research/lstm_object_detection/models/__init__.py 1 research/lstm_object_detection/models/lstm_ssd_interleaved_mobilenet_v2_feature_extractor.py 174 research/lstm_object_detection/models/lstm_ssd_mobilenet_v1_feature_extractor.py 121 research/lstm_object_detection/models/mobilenet_defs.py 93 research/lstm_object_detection/protos/__init__.py 1 research/lstm_object_detection/protos/input_reader_google.proto 24 research/lstm_object_detection/protos/pipeline.proto 53 research/lstm_object_detection/protos/quant_overrides.proto 30 research/lstm_object_detection/tflite/mobile_lstd_tflite_client.cc 183 research/lstm_object_detection/tflite/mobile_lstd_tflite_client.h 39 research/lstm_object_detection/tflite/mobile_ssd_client.cc 166 research/lstm_object_detection/tflite/mobile_ssd_client.h 89 research/lstm_object_detection/tflite/mobile_ssd_tflite_client.cc 456 research/lstm_object_detection/tflite/mobile_ssd_tflite_client.h 62 research/lstm_object_detection/tflite/protos/anchor_generation_options.proto 37 research/lstm_object_detection/tflite/protos/box_encodings.proto 81 research/lstm_object_detection/tflite/protos/detections.proto 31 research/lstm_object_detection/tflite/protos/labelmap.proto 57 research/lstm_object_detection/tflite/protos/mobile_ssd_client_options.proto 92 research/lstm_object_detection/tflite/utils/conversion_utils.cc 41 research/lstm_object_detection/tflite/utils/conversion_utils.h 12 research/lstm_object_detection/tflite/utils/file_utils.cc 34 research/lstm_object_detection/tflite/utils/file_utils.h 14 research/lstm_object_detection/tflite/utils/ssd_utils.cc 463 research/lstm_object_detection/tflite/utils/ssd_utils.h 62 research/lstm_object_detection/train.py 120 research/lstm_object_detection/trainer.py 245 research/lstm_object_detection/utils/__init__.py 1 research/lstm_object_detection/utils/config_util.py 43 research/marco/Automated_Marco.py 30 research/marco/jpeg2json.py 7 research/object_detection/__init__.py 1 research/object_detection/anchor_generators/__init__.py 1 research/object_detection/anchor_generators/flexible_grid_anchor_generator.py 44 research/object_detection/anchor_generators/grid_anchor_generator.py 83 research/object_detection/anchor_generators/multiple_grid_anchor_generator.py 155 research/object_detection/anchor_generators/multiscale_grid_anchor_generator.py 61 research/object_detection/box_coders/__init__.py 1 research/object_detection/box_coders/faster_rcnn_box_coder.py 48 research/object_detection/box_coders/keypoint_box_coder.py 82 research/object_detection/box_coders/mean_stddev_box_coder.py 22 research/object_detection/box_coders/square_box_coder.py 48 research/object_detection/builders/__init__.py 1 research/object_detection/builders/anchor_generator_builder.py 84 research/object_detection/builders/box_coder_builder.py 35 research/object_detection/builders/box_predictor_builder.py 586 research/object_detection/builders/calibration_builder.py 98 research/object_detection/builders/dataset_builder.py 135 research/object_detection/builders/decoder_builder.py 39 research/object_detection/builders/graph_rewriter_builder.py 22 research/object_detection/builders/hyperparams_builder.py 215 research/object_detection/builders/image_resizer_builder.py 106 research/object_detection/builders/input_reader_builder.py 47 research/object_detection/builders/losses_builder.py 154 research/object_detection/builders/matcher_builder.py 26 research/object_detection/builders/model_builder.py 991 research/object_detection/builders/optimizer_builder.py 125 research/object_detection/builders/post_processing_builder.py 73 research/object_detection/builders/preprocessor_builder.py 334 research/object_detection/builders/region_similarity_calculator_builder.py 22 research/object_detection/builders/target_assigner_builder.py 13 research/object_detection/core/__init__.py 1 research/object_detection/core/anchor_generator.py 37 research/object_detection/core/balanced_positive_negative_sampler.py 117 research/object_detection/core/batcher.py 50 research/object_detection/core/box_coder.py 43 research/object_detection/core/box_list.py 59 research/object_detection/core/box_list_ops.py 516 research/object_detection/core/box_predictor.py 50 research/object_detection/core/data_decoder.py 10 research/object_detection/core/data_parser.py 10 research/object_detection/core/densepose_ops.py 156 research/object_detection/core/freezable_batch_norm.py 9 research/object_detection/core/freezable_sync_batch_norm.py 10 research/object_detection/core/keypoint_ops.py 121 research/object_detection/core/losses.py 329 research/object_detection/core/matcher.py 71 research/object_detection/core/minibatch_sampler.py 24 research/object_detection/core/model.py 176 research/object_detection/core/post_processing.py 698 research/object_detection/core/prefetcher.py 16 research/object_detection/core/preprocessor.py 2325 research/object_detection/core/preprocessor_cache.py 51 research/object_detection/core/region_similarity_calculator.py 52 research/object_detection/core/standard_fields.py 144 research/object_detection/core/target_assigner.py 1223 research/object_detection/data_decoders/__init__.py 1 research/object_detection/data_decoders/tf_example_decoder.py 667 research/object_detection/data_decoders/tf_sequence_example_decoder.py 164 research/object_detection/dataset_tools/__init__.py 1 research/object_detection/dataset_tools/context_rcnn/__init__.py 1 research/object_detection/dataset_tools/context_rcnn/add_context_to_examples.py 666 research/object_detection/dataset_tools/context_rcnn/create_cococameratraps_tfexample_main.py 216 research/object_detection/dataset_tools/context_rcnn/generate_detection_data.py 156 research/object_detection/dataset_tools/context_rcnn/generate_embedding_data.py 225 research/object_detection/dataset_tools/create_ava_actions_tf_record.py 373 research/object_detection/dataset_tools/create_coco_tf_record.py 385 research/object_detection/dataset_tools/create_kitti_tf_record.py 171 research/object_detection/dataset_tools/create_oid_tf_record.py 67 research/object_detection/dataset_tools/create_pascal_tf_record.py 119 research/object_detection/dataset_tools/create_pet_tf_record.py 213 research/object_detection/dataset_tools/oid_hierarchical_labels_expansion.py 120 research/object_detection/dataset_tools/oid_tfrecord_creation.py 77 research/object_detection/dataset_tools/seq_example_util.py 159 research/object_detection/dataset_tools/tf_record_creation_util.py 16 research/object_detection/eval_util.py 671 research/object_detection/export_inference_graph.py 95 research/object_detection/export_tflite_graph_lib_tf2.py 198 research/object_detection/export_tflite_graph_tf2.py 54 research/object_detection/export_tflite_ssd_graph.py 43 research/object_detection/export_tflite_ssd_graph_lib.py 202 research/object_detection/exporter.py 421 research/object_detection/exporter_lib_v2.py 186 research/object_detection/exporter_main_v2.py 57 research/object_detection/inference/__init__.py 1 research/object_detection/inference/detection_inference.py 68 research/object_detection/inference/infer_detections.py 52 research/object_detection/inputs.py 677 research/object_detection/legacy/__init__.py 1 research/object_detection/legacy/eval.py 85 research/object_detection/legacy/evaluator.py 185 research/object_detection/legacy/train.py 118 research/object_detection/legacy/trainer.py 250 research/object_detection/matchers/__init__.py 1 research/object_detection/matchers/argmax_matcher.py 89 research/object_detection/matchers/bipartite_matcher.py 22 research/object_detection/matchers/hungarian_matcher.py 19 research/object_detection/meta_architectures/__init__.py 1 research/object_detection/meta_architectures/center_net_meta_arch.py 2385 research/object_detection/meta_architectures/context_rcnn_lib.py 142 research/object_detection/meta_architectures/context_rcnn_lib_tf2.py 119 research/object_detection/meta_architectures/context_rcnn_meta_arch.py 259 research/object_detection/meta_architectures/deepmac_meta_arch.py 788 research/object_detection/meta_architectures/faster_rcnn_meta_arch.py 1456 research/object_detection/meta_architectures/faster_rcnn_meta_arch_test_lib.py 1797 research/object_detection/meta_architectures/rfcn_meta_arch.py 150 research/object_detection/meta_architectures/ssd_meta_arch.py 700 research/object_detection/meta_architectures/ssd_meta_arch_test_lib.py 193 research/object_detection/metrics/__init__.py 1 research/object_detection/metrics/calibration_evaluation.py 109 research/object_detection/metrics/calibration_metrics.py 48 research/object_detection/metrics/coco_evaluation.py 1137 research/object_detection/metrics/coco_tools.py 458 research/object_detection/metrics/io_utils.py 8 research/object_detection/metrics/lvis_evaluation.py 297 research/object_detection/metrics/lvis_tools.py 108 research/object_detection/metrics/offline_eval_map_corloc.py 87 research/object_detection/metrics/oid_challenge_evaluation.py 79 research/object_detection/metrics/oid_challenge_evaluation_utils.py 93 research/object_detection/metrics/oid_vrd_challenge_evaluation.py 85 research/object_detection/metrics/oid_vrd_challenge_evaluation_utils.py 53 research/object_detection/metrics/tf_example_parser.py 79 research/object_detection/model_hparams.py 13 research/object_detection/model_lib.py 761 research/object_detection/model_lib_v2.py 632 research/object_detection/model_main.py 79 research/object_detection/model_main_tf2.py 76 research/object_detection/model_tpu_main.py 97 research/object_detection/models/__init__.py 1 research/object_detection/models/bidirectional_feature_pyramid_generators.py 238 research/object_detection/models/center_net_hourglass_feature_extractor.py 48 research/object_detection/models/center_net_mobilenet_v2_feature_extractor.py 59 research/object_detection/models/center_net_mobilenet_v2_fpn_feature_extractor.py 88 research/object_detection/models/center_net_resnet_feature_extractor.py 68 research/object_detection/models/center_net_resnet_v1_fpn_feature_extractor.py 105 research/object_detection/models/embedded_ssd_mobilenet_v1_feature_extractor.py 73 research/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor.py 94 research/object_detection/models/faster_rcnn_inception_resnet_v2_keras_feature_extractor.py 62 research/object_detection/models/faster_rcnn_inception_v2_feature_extractor.py 155 research/object_detection/models/faster_rcnn_mobilenet_v1_feature_extractor.py 103 research/object_detection/models/faster_rcnn_nas_feature_extractor.py 155 research/object_detection/models/faster_rcnn_pnas_feature_extractor.py 142 research/object_detection/models/faster_rcnn_resnet_keras_feature_extractor.py 122 research/object_detection/models/faster_rcnn_resnet_v1_feature_extractor.py 114 research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py 237 research/object_detection/models/feature_map_generators.py 446 research/object_detection/models/keras_models/__init__.py 1 research/object_detection/models/keras_models/base_models/original_mobilenet_v2.py 251 research/object_detection/models/keras_models/convert_keras_models.py 46 research/object_detection/models/keras_models/hourglass_network.py 339 research/object_detection/models/keras_models/inception_resnet_v2.py 72 research/object_detection/models/keras_models/mobilenet_v1.py 133 research/object_detection/models/keras_models/mobilenet_v2.py 129 research/object_detection/models/keras_models/model_utils.py 16 research/object_detection/models/keras_models/nonlocal_block.py 73 research/object_detection/models/keras_models/resnet_v1.py 249 research/object_detection/models/ssd_efficientnet_bifpn_feature_extractor.py 431 research/object_detection/models/ssd_inception_v2_feature_extractor.py 65 research/object_detection/models/ssd_inception_v3_feature_extractor.py 64 research/object_detection/models/ssd_mobiledet_feature_extractor.py 426 research/object_detection/models/ssd_mobilenet_edgetpu_feature_extractor.py 29 research/object_detection/models/ssd_mobilenet_edgetpu_feature_extractor_testbase.py 84 research/object_detection/models/ssd_mobilenet_v1_feature_extractor.py 67 research/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor.py 116 research/object_detection/models/ssd_mobilenet_v1_fpn_keras_feature_extractor.py 147 research/object_detection/models/ssd_mobilenet_v1_keras_feature_extractor.py 84 research/object_detection/models/ssd_mobilenet_v1_ppn_feature_extractor.py 37 research/object_detection/models/ssd_mobilenet_v2_feature_extractor.py 67 research/object_detection/models/ssd_mobilenet_v2_fpn_feature_extractor.py 115 research/object_detection/models/ssd_mobilenet_v2_fpn_keras_feature_extractor.py 141 research/object_detection/models/ssd_mobilenet_v2_keras_feature_extractor.py 83 research/object_detection/models/ssd_mobilenet_v2_mnasfpn_feature_extractor.py 267 research/object_detection/models/ssd_mobilenet_v3_feature_extractor.py 130 research/object_detection/models/ssd_mobilenet_v3_feature_extractor_testbase.py 84 research/object_detection/models/ssd_pnasnet_feature_extractor.py 80 research/object_detection/models/ssd_resnet_v1_fpn_feature_extractor.py 211 research/object_detection/models/ssd_resnet_v1_fpn_feature_extractor_testbase.py 152 research/object_detection/models/ssd_resnet_v1_fpn_keras_feature_extractor.py 244 research/object_detection/models/ssd_resnet_v1_ppn_feature_extractor.py 128 research/object_detection/models/ssd_resnet_v1_ppn_feature_extractor_testbase.py 57 research/object_detection/models/ssd_spaghettinet_feature_extractor.py 770 research/object_detection/packages/tf1/setup.py 24 research/object_detection/packages/tf2/setup.py 39 research/object_detection/predictors/__init__.py 1 research/object_detection/predictors/convolutional_box_predictor.py 244 research/object_detection/predictors/convolutional_keras_box_predictor.py 271 research/object_detection/predictors/heads/__init__.py 1 research/object_detection/predictors/heads/box_head.py 132 research/object_detection/predictors/heads/class_head.py 165 research/object_detection/predictors/heads/head.py 14 research/object_detection/predictors/heads/keras_box_head.py 171 research/object_detection/predictors/heads/keras_class_head.py 195 research/object_detection/predictors/heads/keras_mask_head.py 238 research/object_detection/predictors/heads/keypoint_head.py 46 research/object_detection/predictors/heads/mask_head.py 184 research/object_detection/predictors/mask_rcnn_box_predictor.py 53 research/object_detection/predictors/mask_rcnn_keras_box_predictor.py 47 research/object_detection/predictors/rfcn_box_predictor.py 81 research/object_detection/predictors/rfcn_keras_box_predictor.py 118 research/object_detection/protos/__init__.py 1 research/object_detection/protos/anchor_generator.proto 16 research/object_detection/protos/argmax_matcher.proto 22 research/object_detection/protos/bipartite_matcher.proto 9 research/object_detection/protos/box_coder.proto 16 research/object_detection/protos/box_predictor.proto 159 research/object_detection/protos/calibration.proto 73 research/object_detection/protos/center_net.proto 426 research/object_detection/protos/eval.proto 133 research/object_detection/protos/faster_rcnn.proto 204 research/object_detection/protos/faster_rcnn_box_coder.proto 13 research/object_detection/protos/flexible_grid_anchor_generator.proto 21 research/object_detection/protos/fpn.proto 39 research/object_detection/protos/graph_rewriter.proto 21 research/object_detection/protos/grid_anchor_generator.proto 24 research/object_detection/protos/hyperparams.proto 121 research/object_detection/protos/image_resizer.proto 82 research/object_detection/protos/input_reader.proto 169 research/object_detection/protos/keypoint_box_coder.proto 14 research/object_detection/protos/losses.proto 198 research/object_detection/protos/matcher.proto 12 research/object_detection/protos/mean_stddev_box_coder.proto 8 research/object_detection/protos/model.proto 21 research/object_detection/protos/multiscale_anchor_generator.proto 19 research/object_detection/protos/optimizer.proto 83 research/object_detection/protos/pipeline.proto 19 research/object_detection/protos/post_processing.proto 74 research/object_detection/protos/preprocessor.proto 522 research/object_detection/protos/region_similarity_calculator.proto 26 research/object_detection/protos/square_box_coder.proto 11 research/object_detection/protos/ssd.proto 160 research/object_detection/protos/ssd_anchor_generator.proto 41 research/object_detection/protos/string_int_label_map.proto 48 research/object_detection/protos/target_assigner.proto 11 research/object_detection/protos/train.proto 131 research/object_detection/tpu_exporters/__init__.py 1 research/object_detection/tpu_exporters/export_saved_model_tpu.py 32 research/object_detection/tpu_exporters/export_saved_model_tpu_lib.py 122 research/object_detection/tpu_exporters/faster_rcnn.py 133 research/object_detection/tpu_exporters/ssd.py 101 research/object_detection/tpu_exporters/utils.py 17 research/object_detection/utils/__init__.py 1 research/object_detection/utils/autoaugment_utils.py 409 research/object_detection/utils/bifpn_utils.py 180 research/object_detection/utils/category_util.py 24 research/object_detection/utils/colab_utils.py 314 research/object_detection/utils/config_util.py 622 research/object_detection/utils/context_manager.py 11 research/object_detection/utils/dataset_util.py 33 research/object_detection/utils/json_utils.py 22 research/object_detection/utils/label_map_util.py 169 research/object_detection/utils/learning_schedules.py 126 research/object_detection/utils/metrics.py 74 research/object_detection/utils/model_util.py 43 research/object_detection/utils/np_box_list.py 48 research/object_detection/utils/np_box_list_ops.py 231 research/object_detection/utils/np_box_mask_list.py 19 research/object_detection/utils/np_box_mask_list_ops.py 152 research/object_detection/utils/np_box_ops.py 31 research/object_detection/utils/np_mask_ops.py 34 research/object_detection/utils/object_detection_evaluation.py 718 research/object_detection/utils/ops.py 530 research/object_detection/utils/patch_ops.py 34 research/object_detection/utils/per_image_evaluation.py 355 research/object_detection/utils/per_image_vrd_evaluation.py 83 research/object_detection/utils/shape_utils.py 170 research/object_detection/utils/spatial_transform_ops.py 222 research/object_detection/utils/static_shape.py 20 research/object_detection/utils/target_assigner_utils.py 226 research/object_detection/utils/tf_version.py 5 research/object_detection/utils/variables_helper.py 102 research/object_detection/utils/visualization_utils.py 851 research/object_detection/utils/vrd_evaluation.py 268 research/pcl_rl/baseline.py 134 research/pcl_rl/controller.py 329 research/pcl_rl/env_spec.py 148 research/pcl_rl/expert_paths.py 119 research/pcl_rl/full_episode_objective.py 43 research/pcl_rl/gym_wrapper.py 75 research/pcl_rl/model.py 241 research/pcl_rl/objective.py 258 research/pcl_rl/optimizers.py 126 research/pcl_rl/policy.py 323 research/pcl_rl/replay_buffer.py 104 research/pcl_rl/trainer.py 390 research/pcl_rl/trust_region.py 123 research/rebar/config.py 7 research/rebar/datasets.py 43 research/rebar/download_data.py 52 research/rebar/logger.py 10 research/rebar/rebar.py 766 research/rebar/rebar_train.py 148 research/rebar/utils.py 82 research/seq_flow_lite/export_to_tflite.py 48 research/seq_flow_lite/input_fn_reader.py 47 research/seq_flow_lite/layers/base_layers.py 91 research/seq_flow_lite/layers/conv_layers.py 80 research/seq_flow_lite/layers/dense_layers.py 72 research/seq_flow_lite/layers/misc_layers.py 52 research/seq_flow_lite/layers/normalization_layers.py 96 research/seq_flow_lite/layers/projection_layers.py 85 research/seq_flow_lite/layers/qrnn_layers.py 330 research/seq_flow_lite/layers/quantization_layers.py 64 research/seq_flow_lite/metric_functions.py 21 research/seq_flow_lite/models/pqrnn.py 62 research/seq_flow_lite/models/prado.py 148 research/seq_flow_lite/models/sgnn/run_tflite.py 31 research/seq_flow_lite/models/sgnn/sgnn.py 101 research/seq_flow_lite/models/sgnn/sgnn_projection.cc 91 research/seq_flow_lite/models/sgnn/sgnn_projection.h 11 research/seq_flow_lite/models/sgnn/sgnn_projection_op_resolver.cc 13 research/seq_flow_lite/models/sgnn/sgnn_projection_op_resolver.h 11 research/seq_flow_lite/models/sgnn/train.py 64 research/seq_flow_lite/tf_ops/build_def.bzl 79 research/seq_flow_lite/tf_ops/projection_normalizer_util.cc 112 research/seq_flow_lite/tf_ops/projection_normalizer_util.h 23 research/seq_flow_lite/tf_ops/projection_tokenizer_util.cc 68 research/seq_flow_lite/tf_ops/projection_tokenizer_util.h 24 research/seq_flow_lite/tf_ops/projection_util.cc 351 research/seq_flow_lite/tf_ops/projection_util.h 84 research/seq_flow_lite/tf_ops/repo.bzl 312 research/seq_flow_lite/tf_ops/sequence_string_projection.cc 325 research/seq_flow_lite/tf_ops/sequence_string_projection_op_v2.cc 197 research/seq_flow_lite/tf_ops/text_distorter.cc 25 research/seq_flow_lite/tf_ops/text_distorter.h 22 research/seq_flow_lite/tf_ops/tf_custom_ops.cc 68 research/seq_flow_lite/tflite_ops/expected_value.cc 98 research/seq_flow_lite/tflite_ops/expected_value.h 11 research/seq_flow_lite/tflite_ops/layer_norm.cc 294 research/seq_flow_lite/tflite_ops/layer_norm.h 11 research/seq_flow_lite/tflite_ops/quantization_util.h 24 research/seq_flow_lite/tflite_ops/registerer.cc 34 research/seq_flow_lite/tflite_ops/sequence_string_projection.cc 393 research/seq_flow_lite/tflite_ops/sequence_string_projection.h 14 research/seq_flow_lite/tflite_ops/tf_tflite_diff_test_util.cc 322 research/seq_flow_lite/tflite_ops/tf_tflite_diff_test_util.h 68 research/seq_flow_lite/trainer.py 123 research/seq_flow_lite/trainer_v2.py 73 research/seq_flow_lite/utils/misc_utils.py 30 research/seq_flow_lite/utils/tflite_utils.py 87 research/slim/__init__.py 1 research/slim/datasets/__init__.py 1 research/slim/datasets/build_imagenet_data.py 295 research/slim/datasets/cifar10.py 45 research/slim/datasets/dataset_factory.py 23 research/slim/datasets/dataset_utils.py 91 research/slim/datasets/download_and_convert_cifar10.py 99 research/slim/datasets/download_and_convert_flowers.py 110 research/slim/datasets/download_and_convert_mnist.py 111 research/slim/datasets/download_and_convert_visualwakewords.py 55 research/slim/datasets/download_and_convert_visualwakewords_lib.py 162 research/slim/datasets/flowers.py 45 research/slim/datasets/imagenet.py 101 research/slim/datasets/mnist.py 45 research/slim/datasets/preprocess_imagenet_validation_data.py 28 research/slim/datasets/process_bounding_boxes.py 122 research/slim/datasets/visualwakewords.py 66 research/slim/deployment/__init__.py 1 research/slim/deployment/model_deploy.py 286 research/slim/download_and_convert_data.py 47 research/slim/eval_image_classifier.py 125 research/slim/export_inference_graph.py 85 research/slim/nets/__init__.py 1 research/slim/nets/alexnet.py 65 research/slim/nets/cifarnet.py 58 research/slim/nets/cyclegan.py 142 research/slim/nets/dcgan.py 109 research/slim/nets/i3d.py 79 research/slim/nets/i3d_utils.py 145 research/slim/nets/inception.py 18 research/slim/nets/inception_resnet_v2.py 249 research/slim/nets/inception_utils.py 35 research/slim/nets/inception_v1.py 242 research/slim/nets/inception_v2.py 422 research/slim/nets/inception_v3.py 382 research/slim/nets/inception_v4.py 221 research/slim/nets/lenet.py 34 research/slim/nets/mobilenet/__init__.py 1 research/slim/nets/mobilenet/conv_blocks.py 269 research/slim/nets/mobilenet/mobilenet.py 227 research/slim/nets/mobilenet/mobilenet_v2.py 132 research/slim/nets/mobilenet/mobilenet_v3.py 275 research/slim/nets/mobilenet_v1.py 196 research/slim/nets/mobilenet_v1_eval.py 87 research/slim/nets/mobilenet_v1_train.py 129 research/slim/nets/nasnet/__init__.py 1 research/slim/nets/nasnet/nasnet.py 383 research/slim/nets/nasnet/nasnet_utils.py 360 research/slim/nets/nasnet/pnasnet.py 193 research/slim/nets/nets_factory.py 117 research/slim/nets/overfeat.py 64 research/slim/nets/pix2pix.py 134 research/slim/nets/post_training_quantization.py 118 research/slim/nets/resnet_utils.py 84 research/slim/nets/resnet_v1.py 213 research/slim/nets/resnet_v2.py 162 research/slim/nets/s3dg.py 380 research/slim/nets/vgg.py 148 research/slim/preprocessing/__init__.py 1 research/slim/preprocessing/cifarnet_preprocessing.py 64 research/slim/preprocessing/inception_preprocessing.py 162 research/slim/preprocessing/lenet_preprocessing.py 18 research/slim/preprocessing/preprocessing_factory.py 55 research/slim/preprocessing/vgg_preprocessing.py 157 research/slim/setup.py 14 research/slim/train_image_classifier.py 382 research/vid2depth/dataset/__init__.py 1 research/vid2depth/dataset/dataset_loader.py 491 research/vid2depth/dataset/gen_data.py 128 research/vid2depth/inference.py 109 research/vid2depth/model.py 291 research/vid2depth/nets.py 133 research/vid2depth/ops/__init__.py 1 research/vid2depth/ops/icp_grad.py 14 research/vid2depth/ops/icp_op.py 11 research/vid2depth/ops/icp_op_kernel.cc 234 research/vid2depth/ops/icp_train_demo.py 137 research/vid2depth/ops/icp_util.py 98 research/vid2depth/ops/pcl_demo.cc 111 research/vid2depth/project.py 174 research/vid2depth/reader.py 164 research/vid2depth/repo.bzl 34 research/vid2depth/train.py 117 research/vid2depth/util.py 67 tensorflow_models/__init__.py 6 tensorflow_models/nlp/__init__.py 4 tensorflow_models/vision/__init__.py 2