GitLab

* Make codes more pythonic.

* Restore the indents

Restore the indents.

Fixed Bug regarding tfrecord shuffling in object_detection

Add lamberta to CODEOWNERS/samples

Add @lamberta to CODEOWNERS/samples

Object detection export dir

* Update the wide_deep code for latest benchmark config.

* Also update the transformer benchmark code.

Release ssdlite mobilenet v2 coco trained model, add quantized training and minor fixes.

* Move the benchmark_uploader to new location.

* Update benchmark logger to streaming upload.

* Fix lint and unit test error.

* delint.

* Update the benchmark uploader test.

Skip the import of benchmark_uploader when bigquery is not installed.

* Merge the 2 classes of benchmark uploader into 1.

* Address review comments.

* delint.

* Execute bigquery upload in a separate thread.

* Change to use python six.moves for importing.

* Address review comments and delint.

* Address review comment.

Adding comment for potential performance impact for model on CPU.

* Fix random failure on py3.

* Fix the order of flag saver to avoid the randomness.

The test is broken when the benchmark_logger_type is set first, and
validated when the benchmark_log_dir is not set yet.

Export @195097388.

Update mobilenet_v1.md

196161788  by Zhichao Lu:

    Add eval_on_train_steps parameter.

    Since the number of samples in train dataset is usually different to the number of samples in the eval dataset.

--
196151742  by Zhichao Lu:

    Add an optional random sampling process for SSD meta arch and update mean stddev coder to use default std dev when corresponding tensor is not added to boxlist field.

--
196148940  by Zhichao Lu:

    Release ssdlite mobilenet v2 coco trained model.

--
196058528  by Zhichao Lu:

    Apply FPN feature map generation before we add additional layers on top of resnet feature extractor.

--
195818367  by Zhichao Lu:

    Add support for exporting detection keypoints.

--
195745420  by Zhichao Lu:

    Introduce include_metrics_per_category option to Object Detection eval_config.

--
195734733  by Zhichao Lu:

    Rename SSDLite config to be more explicit.

--
195717383  by Zhichao Lu:

    Add quantized training to object_detection.

--
195683542  by...

The final BN and ReLU layer is only need for v2 model since it was
doing preactivation in each block.

Addressing reviewers' comments. Added correct handling of order=np.inf.

* Add benchmark logging for wide_deep.

* Fix lint error.

Update mnist_eager.py to use GPU if available and not specified otherwise. At the moment it uses the CPU even if a GPU is available, since "if flags_obj.no_gpu or tf.test.is_gpu_available():" in "def run_mnist_eager(flags_obj):" evaluates to True if a GPU is available.

* forbid resnet v1 fp16

* address PR comments

Updates to differential_privacy/pate following the Banff workshop.

Collapse license cell

Fix typo in iris_data.py

* Add run individual step only option

* Fix comments and update readme

* Add valiation argument

* Address comments

* Make code shorter

* Fix more lints

* rename --version flag and fix tests to correctly specify version rather than verbosity

* rename version to resnet_version throughout

* fix bugs

* delint

* missed layer_test

* fix indent

* Add dataset info and hyper parameter logging for benchmark.

* Address review comments.

* Address the view comment for data schema name.

* Fix test cases.

* Lint fix.

Internal changes to object detection

195269567  by Zhichao Lu:

    Removing image summaries during train mode.

--
195147413  by Zhichao Lu:

    SSDLite config for mobilenet v2.

--
194883585  by Zhichao Lu:

    Simplify TPU compatible nearest neighbor upsampling using reshape and broadcasting.

--
194851009  by Zhichao Lu:

    Include ava v2.1 detection models in model zoo.

--
194292198  by Zhichao Lu:

    Add option to evaluate any checkpoint (without requiring write access to that directory and overwriting any existing logs there).

--
194122420  by Zhichao Lu:

    num_gt_boxes_per_image and num_det_boxes_per_image value incorrect.
    Should be not the expand dim.

--
193974479  by Zhichao Lu:

    Fixing a bug in the coco evaluator.

--
193959861  by Zhichao Lu:

    Read the default batch size from config file.

--
193737238  by Zhichao Lu:

    Fix data augmentation functions.

--
193576336  by Zhichao Lu:

    Add support for training keypoints.

--
193409179  by Zhichao Lu:

    Update protobuf requirements to 3+ in installation docs.

--
193382651  by Zhichao Lu:

    Updating coco evaluation metrics to allow for a batch of image info, rather than a single image.

--
193244778  by Zhichao Lu:

    Remove deprecated batch_norm_trainable field from ssd mobilenet v2 config

--
193228972  by Zhichao Lu:

    Make sure the final layers are also resized proportional to conv_depth_ratio.

--
193204364  by Zhichao Lu:

    Do not add batch norm parameters to final conv2d ops that predict boxes encodings and class scores in weight shared conv box predictor.

    This allows us to set proper bias and force initial predictions to be background when using focal loss.

--
193137342  by Zhichao Lu:

    Add a util function to visualize value histogram as a tf.summary.image.

--
193119411  by Zhichao Lu:

    Adding support for reading in logits as groundtruth labels and applying an optional temperature (scaling) before softmax in support of distillation.

--
193087707  by Zhichao Lu:

    Post-process now works again in train mode.

--
193067658  by Zhichao Lu:

    fix flakiness in testSSDRandomCropWithMultiClassScores due to randomness.

--
192922089  by Zhichao Lu:

    Add option to set dropout for classification net in weight shared box predictor.

--
192850747  by Zhichao Lu:

    Remove inaccurate caveat from proto file.

--
192837477  by Zhichao Lu:

    Extend to accept different ratios of conv channels.

--
192813444  by Zhichao Lu:

    Adding option for one_box_for_all_classes to the box_predictor

--
192624207  by Zhichao Lu:

    Update to trainer to allow for reading multiclass scores

--
192583425  by Zhichao Lu:

    Contains implementation of Visual Relations Detection evaluation metric (per
    image evaluation).

--
192529600  by Zhichao Lu:

    Modify the ssd meta arch to allow the option of not adding an implicit background class.

--
192512429  by Zhichao Lu:

    Refactor model_tpu_main.py files and move continuous eval loop into model_lib.py

--
192494267  by Zhichao Lu:

    Update create_pascal_tf_record.py and create_pet_tf_record.py

--
192485456  by Zhichao Lu:

    Enforcing that all eval metric ops have valid python strings.

--
192472546  by Zhichao Lu:

    Set regularize_depthwise to true in mobilenet_v1_argscope.

--
192421843  by Zhichao Lu:

    Refactoring of Mask-RCNN to put all mask prediction code in third stage.

--
192320460  by Zhichao Lu:

    Returning eval_on_train_input_fn from create_estimator_and_inputs(), rather than using train_input_fn in EVAL mode (which will still have data augmentation).

--
192226678  by Zhichao Lu:

    Access TPUEstimator and CrossShardOptimizer from tf namesspace.

--
192195514  by Zhichao Lu:

    Fix test that was flaky due to randomness

--
192166224  by Zhichao Lu:

    Minor fixes to match git repo.

--
192147130  by Zhichao Lu:

    use shape utils for assertion in feature extractor.

--
192132440  by Zhichao Lu:

    Class agnostic masks for mask_rcnn

--
192006190  by Zhichao Lu:

    Add learning rate summary in EVAL mode in model.py

--
192004845  by Zhichao Lu:

    Migrating away from Experiment class, as it is now deprecated. Also, refactoring into a separate model library and binaries.

--
191957195  by Zhichao Lu:

    Add classification_loss and localiztion_loss metrics for TPU jobs.

--
191932855  by Zhichao Lu:

    Add an option to skip the last striding in mobilenet. The modified network has nominal output stride 16 instead of 32.

--
191787921  by Zhichao Lu:

    Add option to override base feature extractor hyperparams in SSD models. This would allow us to use the same set of hyperparams for the complete feature extractor (base + new layers) if desired.

--
191743097  by Zhichao Lu:

    Adding an attribute to SSD model to indicate which fields in prediction dictionary have a batch dimension. This will be useful for future video models.

--
191668425  by Zhichao Lu:

    Internal change.

--
191649512  by Zhichao Lu:

    Introduce two parameters in ssd.proto - freeze_batchnorm, inplace_batchnorm_update - and set up slim arg_scopes in ssd_meta_arch.py such that applies it to all batchnorm ops in the predict() method.

    This centralizes the control of freezing and doing inplace batchnorm updates.

--
191620303  by Zhichao Lu:

    Modifications to the preprocessor to support multiclass scores

--

PiperOrigin-RevId: 195269567

* Revert 823da318. This restores distribution strategies for resnet.

This commit is not a direct revert due to significant merge conflict
resolution.

* fix flags test

* npc is no longer used in resnet