1. 15 Jul, 2019 1 commit
    • pkulzc's avatar
      Object detection changes: (#7208) · fe748d4a
      pkulzc authored
      257914648  by lzc:
      
          Internal changes
      
      --
      257525973  by Zhichao Lu:
      
          Fixes bug that silently prevents checkpoints from loading when training w/ eager + functions. Also sets up scripts to run training.
      
      --
      257296614  by Zhichao Lu:
      
          Adding detection_features to model outputs
      
      --
      257234565  by Zhichao Lu:
      
          Fix wrong order of `classes_with_max_scores` in class-agnostic NMS caused by
          sorting in partitioned-NMS.
      
      --
      257232002  by ronnyvotel:
      
          Supporting `filter_nonoverlapping` option in np_box_list_ops.clip_to_window().
      
      --
      257198282  by Zhichao Lu:
      
          Adding the focal loss and l1 loss from the Objects as Points paper.
      
      --
      257089535  by Zhichao Lu:
      
          Create Keras based ssd + resnetv1 + fpn.
      
      --
      257087407  by Zhichao Lu:
      
          Make object_detection/data_decoders Python3-compatible.
      
      --
      257004582  by Zhichao Lu:
      
          Updates _decode_raw_data_into_masks_and_boxes to the latest binary masks-to-string encoding format.
      
      --
      257002124  by Zhichao Lu:
      
          Make object_detection/utils Python3-compatible, except json_utils.
      
          The patching trick used in json_utils is not going to work in Python 3.
      
      --
      256795056  by lzc:
      
          Add a detection_anchor_indices field to detection outputs.
      
      --
      256477542  by Zhichao Lu:
      
          Make object_detection/core Python3-compatible.
      
      --
      256387593  by Zhichao Lu:
      
          Edit class_id_function_approximations builder to skip class ids not present in label map.
      
      --
      256259039  by Zhichao Lu:
      
          Move NMS to TPU for FasterRCNN.
      
      --
      256071360  by rathodv:
      
          When multiclass_scores is empty, add one-hot encoding of groundtruth_classes as multiclass scores so that data_augmentation ops that expect the presence of multiclass_scores don't have to individually handle this case.
      
          Also copy input tensor_dict to out_tensor_dict first to avoid inplace modification.
      
      --
      256023645  by Zhichao Lu:
      
          Adds the first WIP iterations of TensorFlow v2 eager + functions style custom training & evaluation loops.
      
      --
      255980623  by Zhichao Lu:
      
          Adds a new data augmentation operation "remap_labels" which remaps a set of labels to a new label.
      
      --
      255753259  by Zhichao Lu:
      
          Announcement of the released evaluation tutorial for Open Images Challenge
          2019.
      
      --
      255698776  by lzc:
      
          Fix rewrite_nn_resize_op function which was broken by tf forward compatibility movement.
      
      --
      255623150  by Zhichao Lu:
      
          Add Keras-based ResnetV1 models.
      
      --
      255504992  by Zhichao Lu:
      
          Fixing the typo in specifying label expansion for ground truth segmentation
          file.
      
      --
      255470768  by Zhichao Lu:
      
          1. Fixing Python bug with parsed arguments.
          2. Adding capability to parse relevant columns from CSV header.
          3. Fixing bug with duplicated labels expansion.
      
      --
      255462432  by Zhichao Lu:
      
          Adds a new data augmentation operation "drop_label_probabilistically" which drops a given label with the given probability. This supports experiments on training in the presence of label noise.
      
      --
      255441632  by rathodv:
      
          Fallback on groundtruth classes when multiclass_scores tensor is empty.
      
      --
      255434899  by Zhichao Lu:
      
          Ensuring evaluation binary can run even with big files by synchronizing
          processing of ground truth and predictions: in this way, ground truth is not stored but immediatly
          used for evaluation. In case gt of object masks, this allows to run
          evaluations on relatively large sets.
      
      --
      255337855  by lzc:
      
          Internal change.
      
      --
      255308908  by Zhichao Lu:
      
          Add comment to clarify usage of calibration parameters proto.
      
      --
      255266371  by Zhichao Lu:
      
          Ensuring correct processing of the case, when no groundtruth masks are provided
          for an image.
      
      --
      255236648  by Zhichao Lu:
      
          Refactor model_builder in faster_rcnn.py to a util_map, so that it's possible to be overwritten.
      
      --
      255093285  by Zhichao Lu:
      
          Updating capability to subsample data during evaluation
      
      --
      255081222  by rathodv:
      
          Convert groundtruth masks to be of type float32 before its used in the loss function.
      
          When using mixed precision training, masks are represented using bfloat16 tensors in the input pipeline for performance reasons. We need to convert them to float32 before using it in the loss function.
      
      --
      254788436  by Zhichao Lu:
      
          Add forward_compatible to non_max_suppression_with_scores to make it is
          compatible with older tensorflow version.
      
      --
      254442362  by Zhichao Lu:
      
          Add num_layer field to ssd feature extractor proto.
      
      --
      253911582  by jonathanhuang:
      
          Plumbs Soft-NMS options (using the new tf.image.non_max_suppression_with_scores op) into the TF Object Detection API.  It adds a `soft_nms_sigma` field to the postprocessing proto file and plumbs this through to both the multiclass and class_agnostic versions of NMS. Note that there is no effect on behavior of NMS when soft_nms_sigma=0 (which it is set to by default).
      
          See also "Soft-NMS -- Improving Object Detection With One Line of Code" by Bodla et al (https://arxiv.org/abs/1704.04503)
      
      --
      253703949  by Zhichao Lu:
      
          Internal test fixes.
      
      --
      253151266  by Zhichao Lu:
      
          Fix the op type check for FusedBatchNorm, given that we introduced
          FusedBatchNormV3 in a previous change.
      
      --
      252718956  by Zhichao Lu:
      
          Customize activation function to enable relu6 instead of relu for saliency
          prediction model seastarization
      
      --
      252158593  by Zhichao Lu:
      
          Make object_detection/core Python3-compatible.
      
      --
      252150717  by Zhichao Lu:
      
          Make object_detection/core Python3-compatible.
      
      --
      251967048  by Zhichao Lu:
      
          Make GraphRewriter proto extensible.
      
      --
      251950039  by Zhichao Lu:
      
          Remove experimental_export_device_assignment from TPUEstimator.export_savedmodel(), so as to remove rewrite_for_inference().
      
          As a replacement, export_savedmodel() V2 API supports device_assignment where user call tpu.rewrite in model_fn and pass in device_assigment there.
      
      --
      251890697  by rathodv:
      
          Updated docstring to include new output nodes.
      
      --
      251662894  by Zhichao Lu:
      
          Add autoaugment augmentation option to objection detection api codebase. This
          is an available option in preprocessor.py.
      
          The intended usage of autoaugment is to be done along with random flipping and
          cropping for best results.
      
      --
      251532908  by Zhichao Lu:
      
          Add TrainingDataType enum to track whether class-specific or agnostic data was used to fit the calibration function.
      
          This is useful, since classes with few observations may require a calibration function fit on all classes.
      
      --
      251511339  by Zhichao Lu:
      
          Add multiclass isotonic regression to the calibration builder.
      
      --
      251317769  by pengchong:
      
          Internal Change.
      
      --
      250729989  by Zhichao Lu:
      
          Fixing bug in gt statistics count in case of mask and box annotations.
      
      --
      250729627  by Zhichao Lu:
      
          Label expansion for segmentation.
      
      --
      250724905  by Zhichao Lu:
      
          Fix use_depthwise in fpn and test it with fpnlite on ssd + mobilenet v2.
      
      --
      250670379  by Zhichao Lu:
      
          Internal change
      
      250630364  by lzc:
      
          Fix detection_model_zoo footnotes
      
      --
      250560654  by Zhichao Lu:
      
          Fix static shape issue in matmul_crop_and_resize.
      
      --
      250534857  by Zhichao Lu:
      
          Edit class agnostic calibration function docstring to more accurately describe the function's outputs.
      
      --
      250533277  by Zhichao Lu:
      
          Edit the multiclass messages to use class ids instead of labels.
      
      --
      
      PiperOrigin-RevId: 257914648
      fe748d4a
  2. 10 Jul, 2019 1 commit
  3. 09 Jul, 2019 1 commit
  4. 08 Jul, 2019 2 commits
  5. 26 Jun, 2019 1 commit
  6. 14 Jun, 2019 1 commit
    • André Araujo's avatar
      Some refactoring + Google Landmarks dataset scripts (#7014) · 7cd29f8c
      André Araujo authored
      * Merged commit includes the following changes:
      253126424  by Andre Araujo:
      
          Scripts to compute metrics for Google Landmarks dataset.
      
          Also, a small fix to metric in retrieval case: avoids duplicate predicted images.
      
      --
      253118971  by Andre Araujo:
      
          Metrics for Google Landmarks dataset.
      
      --
      253106953  by Andre Araujo:
      
          Library to read files from Google Landmarks challenges.
      
      --
      250700636  by Andre Araujo:
      
          Handle case of aggregation extraction with empty set of input features.
      
      --
      250516819  by Andre Araujo:
      
          Add minimum size for DELF extractor.
      
      --
      250435822  by Andre Araujo:
      
          Add max_image_size/min_image_size for open-source DELF proto / module.
      
      --
      250414606  by Andre Araujo:
      
          Refactor extract_aggregation to allow reuse with different datasets.
      
      --
      250356863  by Andre Araujo:
      
          Remove unnecessary cmd_args variable from boxes_and_features_extraction.
      
      --
      249783379  by Andre Araujo:
      
          Create directory for writing mapping file if it does not exist.
      
      --
      249581591  by Andre Araujo:
      
          Refactor scripts to extract boxes and features from images in Revisited datasets.
          Also, change tf.logging.info --> print for easier logging in open source code.
      
      --
      249511821  by Andre Araujo:
      
          Small change to function for file/directory handling.
      
      --
      249289499  by Andre Araujo:
      
          Internal change.
      
      --
      
      PiperOrigin-RevId: 253126424
      
      * Updating DELF init to adjust to latest changes
      
      * Editing init files for python packages
      
      * Edit D2R dataset reader to work with py3.
      
      PiperOrigin-RevId: 253135576
      
      * DELF package: fix import ordering
      7cd29f8c
  7. 13 Jun, 2019 1 commit
  8. 12 Jun, 2019 1 commit
  9. 31 May, 2019 2 commits
    • Andrew M Dai's avatar
      Add step 1 instructions for MaskGAN. (#6917) · de10c0c2
      Andrew M Dai authored
      * Add step 1 instructions for MaskGAN.
      de10c0c2
    • pkulzc's avatar
      Merged commit includes the following changes: (#6932) · 9bbf8015
      pkulzc authored
      250447559  by Zhichao Lu:
      
          Update expected files format for Instance Segmentation challenge:
          - add fields ImageWidth, ImageHeight and store the values per prediction
          - as mask, store only encoded image and assume its size is ImageWidth x ImageHeight
      
      --
      250402780  by rathodv:
      
          Fix failing Mask R-CNN TPU convergence test.
      
          Cast second stage prediction tensors from bfloat16 to float32 to prevent errors in third target assignment (Mask Prediction) - Concat with different types bfloat16 and bfloat32 isn't allowed.
      
      --
      250300240  by Zhichao Lu:
      
          Addion Open Images Challenge 2019 object detection and instance segmentation
          support into Estimator framework.
      
      --
      249944839  by rathodv:
      
          Modify exporter.py to add multiclass score nodes in exported inference graphs.
      
      --
      249935201  by rathodv:
      
          Modify postprocess methods to preserve multiclass scores after non max suppression.
      
      --
      249878079  by Zhichao Lu:
      
          This CL slightly refactors some Object Detection helper functions for data creation, evaluation, and groundtruth providing.
      
          This will allow the eager+function custom loops to share code with the existing estimator training loops.
      
          Concretely we make the following changes:
          1. In input creation we separate dataset-creation into top-level helpers, and allow it to optionally accept a pre-constructed model directly instead of always creating a model from the config just for feature preprocessing.
      
          2. In coco evaluation we split the update_op creation into its own function, which the custom loops will call directly.
      
          3. In model_lib we move groundtruth providing/ datastructure munging into a helper function
      
          4. For now we put an escape hatch in `_summarize_target_assignment` when executing in tf v2.0 behavior because the summary apis used only work w/ tf 1.x
      
      --
      249673507  by rathodv:
      
          Use explicit casts instead of tf.to_float and tf.to_int32 to avoid warnings.
      
      --
      249656006  by Zhichao Lu:
      
          Add named "raw_keypoint_locations" node that corresponds with the "raw_box_locations" node.
      
      --
      249651674  by rathodv:
      
          Keep proposal boxes in float format. MatMulCropAndResize can handle the type even when feature themselves are bfloat16s.
      
      --
      249568633  by rathodv:
      
          Support q > 1 in class agnostic NMS.
          Break post_processing_test.py into 3 separate files to avoid linter errors.
      
      --
      249535530  by rathodv:
      
          Update some deprecated arguments to tf ops.
      
      --
      249368223  by rathodv:
      
          Modify MatMulCropAndResize to use MultiLevelRoIAlign method and move the tests to spatial_transform_ops.py module.
      
          This cl establishes that CropAndResize and RoIAlign are equivalent and only differ in the sampling point grid within the boxes. CropAndResize uses a uniform size x size point grid such that the corner points exactly overlap box corners, while RoiAlign divides boxes into size x size cells and uses their centers as sampling points. In this cl, we switch MatMulCropAndResize to use the MultiLevelRoIAlign implementation with `align_corner` option as MultiLevelRoIAlign implementation is more memory efficient on TPU when compared to the original MatMulCropAndResize.
      
      --
      249337338  by chowdhery:
      
          Add class-agnostic non-max-suppression in post_processing
      
      --
      249139196  by Zhichao Lu:
      
          Fix positional argument bug in export_tflite_ssd_graph
      
      --
      249120219  by Zhichao Lu:
      
          Add evaluator for computing precision limited to a given recall range.
      
      --
      249030593  by Zhichao Lu:
      
          Evaluation util to run segmentation and detection challenge evaluation.
      
      --
      248554358  by Zhichao Lu:
      
          This change contains the auxiliary changes required for TF 2.0 style training with eager+functions+dist strat loops, but not the loops themselves.
      
          It includes:
          - Updates to shape usage to support both tensorshape v1 and tensorshape v2
          - A fix to FreezableBatchNorm to not override the `training` arg in call when `None` was passed to the constructor (Not an issue in the estimator loops but it was in the custom loops)
          - Puts some constants in init_scope so they work in eager + functions
          - Makes learning rate schedules return a callable in eager mode (required so they update when the global_step changes)
          - Makes DetectionModel a tf.module so it tracks variables (e.g. ones nested in layers)
          - Removes some references to `op.name` for some losses and replaces it w/ explicit names
          - A small part of the change to allow the coco evaluation metrics to work in eager mode
      
      --
      248271226  by rathodv:
      
          Add MultiLevel RoIAlign op.
      
      --
      248229103  by rathodv:
      
          Add functions to 1. pad features maps 2. ravel 5-D indices
      
      --
      248206769  by rathodv:
      
          Add utilities needed to introduce RoI Align op.
      
      --
      248177733  by pengchong:
      
          Internal changes
      
      --
      247742582  by Zhichao Lu:
      
          Open Images Challenge 2019 instance segmentation metric: part 2
      
      --
      247525401  by Zhichao Lu:
      
          Update comments on max_class_per_detection.
      
      --
      247520753  by rathodv:
      
          Add multilevel crop and resize operation that builds on top of matmul_crop_and_resize.
      
      --
      247391600  by Zhichao Lu:
      
          Open Images Challenge 2019 instance segmentation metric
      
      --
      247325813  by chowdhery:
      
          Quantized MobileNet v2 SSD FPNLite config with depth multiplier 0.75
      
      --
      
      PiperOrigin-RevId: 250447559
      9bbf8015
  10. 29 May, 2019 1 commit
    • Marvin Teichmann's avatar
      Put all python dependencies into one line. (#6870) · e0388cfe
      Marvin Teichmann authored
      * Put all python dependencies into one line.
      
      This makes it easier to copy, paste & install all dependencies at once. In addition many users have custom setups (virtualenv, conda, .etc). Having it in one line easily allows to grap the dependencies.
      
      * Remove 'sudo' from all pip install commands and adjust troubleshooting section.
      e0388cfe
  11. 28 May, 2019 1 commit
  12. 22 May, 2019 2 commits
    • Zhuoran Liu's avatar
      Add TPU SavedModel exporter and refactor OD code (#6737) · 80444539
      Zhuoran Liu authored
      247226201  by ronnyvotel:
      
          Updating the visualization tools to accept unique_ids for color coding.
      
      --
      247067830  by Zhichao Lu:
      
          Add box_encodings_clip_range options for the convolutional box predictor (for TPU compatibility).
      
      --
      246888475  by Zhichao Lu:
      
          Remove unused _update_eval_steps function.
      
      --
      246163259  by lzc:
      
          Add a gather op that can handle ignore indices (which are "-1"s in this case).
      
      --
      246084944  by Zhichao Lu:
      
          Keras based implementation for SSD + MobilenetV2 + FPN.
      
      --
      245544227  by rathodv:
      
          Add batch_get_targets method to target assigner module to gather any groundtruth tensors based on the results of target assigner.
      
      --
      245540854  by rathodv:
      
          Update target assigner to return match tensor instead of a match object.
      
      --
      245434441  by Zhichao Lu:
      
          Add README for tpu_exporters package.
      
      --
      245381834  by lzc:
      
          Internal change.
      
      --
      245298983  by Zhichao Lu:
      
          Add conditional_shape_resizer to config_util
      
      --
      245134666  by Zhichao Lu:
      
          Adds ConditionalShapeResizer to the ImageResizer proto which enables resizing only if input image height or width is is greater or smaller than a certain size. Also enables specification of resize method in resize_to_{max, min}_dimension methods.
      
      --
      245093975  by Zhichao Lu:
      
          Exporting SavedModel for Object Detection TPU inference. (faster-rcnn)
      
      --
      245072421  by Zhichao Lu:
      
          Adds a new image resizing method "resize_to_max_dimension" which resizes images only if a dimension is greater than the maximum desired value while maintaining aspect ratio.
      
      --
      244946998  by lzc:
      
          Internal Changes.
      
      --
      244943693  by Zhichao Lu:
      
          Add a custom config to mobilenet v2 that makes it more detection friendly.
      
      --
      244754158  by derekjchow:
      
          Internal change.
      
      --
      244699875  by Zhichao Lu:
      
          Add check_range=False to box_list_ops.to_normalized_coordinates when training
          for instance segmentation.  This is consistent with other calls when training
          for object detection.  There could be wrongly annotated boxes in the dataset.
      
      --
      244507425  by rathodv:
      
          Support bfloat16 for ssd models.
      
      --
      244399982  by Zhichao Lu:
      
          Exporting SavedModel for Object Detection TPU inference. (ssd)
      
      --
      244209387  by Zhichao Lu:
      
          Internal change.
      
      --
      243922296  by rathodv:
      
          Change `raw_detection_scores` to contain softmax/sigmoid scores (not logits) for `raw_ detection_boxes`.
      
      --
      243883978  by Zhichao Lu:
      
          Add a sample fully conv config.
      
      --
      243369455  by Zhichao Lu:
      
          Fix regularization loss gap in Keras and Slim.
      
      --
      243292002  by lzc:
      
          Internal changes.
      
      --
      243097958  by Zhichao Lu:
      
          Exporting SavedModel for Object Detection TPU inference. (ssd model)
      
      --
      243007177  by Zhichao Lu:
      
          Exporting SavedModel for Object Detection TPU inference. (ssd model)
      
      --
      242776550  by Zhichao Lu:
      
          Make object detection pre-processing run on GPU.  tf.map_fn() uses
          TensorArrayV3 ops, which have no int32 GPU implementation.  Cast to int64,
          then cast back to int32.
      
      --
      242723128  by Zhichao Lu:
      
          Using sorted dictionaries for additional heads in non_max_suppression to ensure tensor order
      
      --
      242495311  by Zhichao Lu:
      
          Update documentation to reflect new TFLite examples repo location
      
      --
      242230527  by Zhichao Lu:
      
          Fix Dropout bugs for WeightSharedConvolutionalBoxPred.
      
      --
      242226573  by Zhichao Lu:
      
          Create Keras-based WeightSharedConvolutionalBoxPredictor.
      
      --
      241806074  by Zhichao Lu:
      
          Add inference in unit tests of TFX OD template.
      
      --
      241641498  by lzc:
      
          Internal change.
      
      --
      241637481  by Zhichao Lu:
      
          matmul_crop_and_resize(): Switch to dynamic shaping, so that not all dimensions are required to be known.
      
      --
      241429980  by Zhichao Lu:
      
          Internal change
      
      --
      241167237  by Zhichao Lu:
      
          Adds a faster_rcnn_inception_resnet_v2 Keras feature extractor, and updates the model builder to construct it.
      
      --
      241088616  by Zhichao Lu:
      
          Make it compatible with different dtype, e.g. float32, bfloat16, etc.
      
      --
      240897364  by lzc:
      
          Use image_np_expanded in object_detection_tutorial notebook.
      
      --
      240890393  by Zhichao Lu:
      
          Disable multicore inference for OD template as its not yet compatible.
      
      --
      240352168  by Zhichao Lu:
      
          Make SSDResnetV1FpnFeatureExtractor not protected to allow inheritance.
      
      --
      240351470  by lzc:
      
          Internal change.
      
      --
      239878928  by Zhichao Lu:
      
          Defines Keras box predictors for Faster RCNN and RFCN
      
      --
      239872103  by Zhichao Lu:
      
          Delete duplicated inputs in test.
      
      --
      239714273  by Zhichao Lu:
      
          Adding scope variable to all class heads
      
      --
      239698643  by Zhichao Lu:
      
          Create FPN feature extractor for object detection.
      
      --
      239696657  by Zhichao Lu:
      
          Internal Change.
      
      --
      239299404  by Zhichao Lu:
      
          Allows the faster rcnn meta-architecture to support Keras subcomponents
      
      --
      238502595  by Zhichao Lu:
      
          Lay the groundwork for symmetric quantization.
      
      --
      238496885  by Zhichao Lu:
      
          Add flexible_grid_anchor_generator
      
      --
      238138727  by lzc:
      
          Remove dead code.
      
          _USE_C_SHAPES has been forced True in TensorFlow releases since
          TensorFlow 1.9
          (https://github.com/tensorflow/tensorflow/commit/1d74a69443f741e69f9f52cb6bc2940b4d4ae3b7)
      
      --
      238123936  by rathodv:
      
          Add num_matched_groundtruth summary to target assigner in SSD.
      
      --
      238103345  by ronnyvotel:
      
          Raising error if input file pattern does not match any files.
          Also printing the number of evaluation images for coco metrics.
      
      --
      238044081  by Zhichao Lu:
      
          Fix docstring to state the correct dimensionality of `class_predictions_with_background`.
      
      --
      237920279  by Zhichao Lu:
      
          [XLA] Rework debug flags for dumping HLO.
      
          The following flags (usually passed via the XLA_FLAGS envvar) are removed:
      
            xla_dump_computations_to
            xla_dump_executions_to
            xla_dump_ir_to
            xla_dump_optimized_hlo_proto_to
            xla_dump_per_pass_hlo_proto_to
            xla_dump_unoptimized_hlo_proto_to
            xla_generate_hlo_graph
            xla_generate_hlo_text_to
            xla_hlo_dump_as_html
            xla_hlo_graph_path
            xla_log_hlo_text
      
          The following new flags are added:
      
            xla_dump_to
            xla_dump_hlo_module_re
            xla_dump_hlo_pass_re
            xla_dump_hlo_as_text
            xla_dump_hlo_as_proto
            xla_dump_hlo_as_dot
            xla_dump_hlo_as_url
            xla_dump_hlo_as_html
            xla_dump_ir
            xla_dump_hlo_snapshots
      
          The default is not to dump anything at all, but as soon as some dumping flag is
          specified, we enable the following defaults (most of which can be overridden).
      
           * dump to stdout (overridden by --xla_dump_to)
           * dump HLO modules at the very beginning and end of the optimization pipeline
           * don't dump between any HLO passes (overridden by --xla_dump_hlo_pass_re)
           * dump all HLO modules (overridden by --xla_dump_hlo_module_re)
           * dump in textual format (overridden by
             --xla_dump_hlo_as_{text,proto,dot,url,html}).
      
          For example, to dump optimized and unoptimized HLO text and protos to /tmp/foo,
          pass
      
            --xla_dump_to=/tmp/foo --xla_dump_hlo_as_text --xla_dump_hlo_as_proto
      
          For details on these flags' meanings, see xla.proto.
      
          The intent of this change is to make dumping both simpler to use and more
          powerful.
      
          For example:
      
           * Previously there was no way to dump the HLO module during the pass pipeline
             in HLO text format; the only option was --dump_per_pass_hlo_proto_to, which
             dumped in proto format.
      
             Now this is --xla_dump_pass_re=.* --xla_dump_hlo_as_text.  (In fact, the
             second flag is not necessary in this case, as dumping as text is the
             default.)
      
           * Previously there was no way to dump HLO as a graph before and after
             compilation; the only option was --xla_generate_hlo_graph, which would dump
             before/after every pass.
      
             Now this is --xla_dump_hlo_as_{dot,url,html} (depending on what format you
             want the graph in).
      
           * Previously, there was no coordination between the filenames written by the
             various flags, so info about one module might be dumped with various
             filename prefixes.  Now the filenames are consistent and all dumps from a
             particular module are next to each other.
      
          If you only specify some of these flags, we try to figure out what you wanted.
          For example:
      
           * --xla_dump_to implies --xla_dump_hlo_as_text unless you specify some
             other --xla_dump_as_* flag.
      
           * --xla_dump_hlo_as_text or --xla_dump_ir implies dumping to stdout unless you
             specify a different --xla_dump_to directory.  You can explicitly dump to
             stdout with --xla_dump_to=-.
      
          As part of this change, I simplified the debugging code in the HLO passes for
          dumping HLO modules.  Previously, many tests explicitly VLOG'ed the HLO module
          before, after, and sometimes during the pass.  I removed these VLOGs.  If you
          want dumps before/during/after an HLO pass, use --xla_dump_pass_re=<pass_name>.
      
      --
      237510043  by lzc:
      
          Internal Change.
      
      --
      237469515  by Zhichao Lu:
      
          Parameterize model_builder.build in inputs.py.
      
      --
      237293511  by rathodv:
      
          Remove multiclass_scores from tensor_dict in transform_data_fn always.
      
      --
      237260333  by ronnyvotel:
      
          Updating faster_rcnn_meta_arch to define prediction dictionary fields that are batched.
      
      --
      
      PiperOrigin-RevId: 247226201
      80444539
    • tjakob's avatar
      Use new tensorrt API (#6828) · d11d9845
      tjakob authored
      d11d9845
  13. 21 May, 2019 3 commits
    • André Araujo's avatar
      Small edits to DELF markdown instructions (#6839) · 76256146
      André Araujo authored
      * Initial feature aggregation code for Detect-to-Retrieve paper.
      
      PiperOrigin-RevId: 246043144
      
      * Add support for ASMK/ASMK*/R-ASMK/R-ASMK*.
      
      PiperOrigin-RevId: 247337028
      
      * Add DatumProto uint32 field, and limit datum_io to uint32 and float32/float64 types.
      
      Also, introduce DatumPairProto, to be used for ASMK variants. Functions to read/write in this new format are added and tested.
      
      PiperOrigin-RevId: 247515205
      
      * Add batching option to feature aggregation extraction.
      
      PiperOrigin-RevId: 247614627
      
      * Script to perform local feature aggregation, with associated configs.
      
      Also small edits to the aggregation extractor, for better handling of input features / avoiding OOM.
      
      PiperOrigin-RevId: 248150750
      
      * Tests to check that aggregation using regions with no local features works.
      
      PiperOrigin-RevId: 248153275
      
      * Include new library/proto for aggregation
      
      * Merged commit includes the following changes:
      
      PiperOrigin-RevId: 248176511
      
      * Merged commit includes the following changes:
      248194572  by Andre Araujo:
      
          Change tf.tensor_scatter_nd_add --> tf.compat.v1.tensor_scatter_add to make it compatible with TF 1.X.
      
      --
      
      PiperOrigin-RevId: 248194572
      
      * Functions to parse ground-truth and compute metrics for revisited datasets.
      
      Unit tests are added.
      
      PiperOrigin-RevId: 248561575
      
      * Small change to argparse bool option, which does not work as expected.
      
      PiperOrigin-RevId: 248805505
      
      * Class to compute similarity between aggregated descriptors.
      
      PiperOrigin-RevId: 249102986
      
      * Script to perform retrieval and compute metrics.
      
      PiperOrigin-RevId: 249104011
      
      * feature_aggregation_similarity library in DELF init
      
      * D2R instructions / README update
      
      * Small edit to README
      
      * Internal change.
      
      PiperOrigin-RevId: 249113531
      
      * Instructions to reproduce D2R paper results, and small edits to config files.
      
      PiperOrigin-RevId: 249159850
      
      * Revert "Internal change."
      
      This reverts commit 0529f2b8471a20c88f1fbb37367f142965d098ee.
      
      Undoing incorrect markdown deletion.
      
      * Small updates to markdown instructions
      
      * Small updates to markdown instructions 2
      76256146
    • André Araujo's avatar
      Undo Detect-to-Retrieve markdown deletion (#6831) · 5ec581ae
      André Araujo authored
      * Initial feature aggregation code for Detect-to-Retrieve paper.
      
      PiperOrigin-RevId: 246043144
      
      * Add support for ASMK/ASMK*/R-ASMK/R-ASMK*.
      
      PiperOrigin-RevId: 247337028
      
      * Add DatumProto uint32 field, and limit datum_io to uint32 and float32/float64 types.
      
      Also, introduce DatumPairProto, to be used for ASMK variants. Functions to read/write in this new format are added and tested.
      
      PiperOrigin-RevId: 247515205
      
      * Add batching option to feature aggregation extraction.
      
      PiperOrigin-RevId: 247614627
      
      * Script to perform local feature aggregation, with associated configs.
      
      Also small edits to the aggregation extractor, for better handling of input features / avoiding OOM.
      
      PiperOrigin-RevId: 248150750
      
      * Tests to check that aggregation using regions with no local features works.
      
      PiperOrigin-RevId: 248153275
      
      * Include new library/proto for aggregation
      
      * Merged commit includes the following changes:
      
      PiperOrigin-RevId: 248176511
      
      * Merged commit includes the following changes:
      248194572  by Andre Araujo:
      
          Change tf.tensor_scatter_nd_add --> tf.compat.v1.tensor_scatter_add to make it compatible with TF 1.X.
      
      --
      
      PiperOrigin-RevId: 248194572
      
      * Functions to parse ground-truth and compute metrics for revisited datasets.
      
      Unit tests are added.
      
      PiperOrigin-RevId: 248561575
      
      * Small change to argparse bool option, which does not work as expected.
      
      PiperOrigin-RevId: 248805505
      
      * Class to compute similarity between aggregated descriptors.
      
      PiperOrigin-RevId: 249102986
      
      * Script to perform retrieval and compute metrics.
      
      PiperOrigin-RevId: 249104011
      
      * feature_aggregation_similarity library in DELF init
      
      * D2R instructions / README update
      
      * Small edit to README
      
      * Internal change.
      
      PiperOrigin-RevId: 249113531
      
      * Instructions to reproduce D2R paper results, and small edits to config files.
      
      PiperOrigin-RevId: 249159850
      
      * Revert "Internal change."
      
      This reverts commit 0529f2b8471a20c88f1fbb37367f142965d098ee.
      
      Undoing incorrect markdown deletion.
      5ec581ae
    • André Araujo's avatar
      Code for Detect-to-Retrieve fully integrated (#6829) · 9062d200
      André Araujo authored
      * Initial feature aggregation code for Detect-to-Retrieve paper.
      
      PiperOrigin-RevId: 246043144
      
      * Add support for ASMK/ASMK*/R-ASMK/R-ASMK*.
      
      PiperOrigin-RevId: 247337028
      
      * Add DatumProto uint32 field, and limit datum_io to uint32 and float32/float64 types.
      
      Also, introduce DatumPairProto, to be used for ASMK variants. Functions to read/write in this new format are added and tested.
      
      PiperOrigin-RevId: 247515205
      
      * Add batching option to feature aggregation extraction.
      
      PiperOrigin-RevId: 247614627
      
      * Script to perform local feature aggregation, with associated configs.
      
      Also small edits to the aggregation extractor, for better handling of input features / avoiding OOM.
      
      PiperOrigin-RevId: 248150750
      
      * Tests to check that aggregation using regions with no local features works.
      
      PiperOrigin-RevId: 248153275
      
      * Include new library/proto for aggregation
      
      * Merged commit includes the following changes:
      
      PiperOrigin-RevId: 248176511
      
      * Merged commit includes the following changes:
      248194572  by Andre Araujo:
      
          Change tf.tensor_scatter_nd_add --> tf.compat.v1.tensor_scatter_add to make it compatible with TF 1.X.
      
      --
      
      PiperOrigin-RevId: 248194572
      
      * Functions to parse ground-truth and compute metrics for revisited datasets.
      
      Unit tests are added.
      
      PiperOrigin-RevId: 248561575
      
      * Small change to argparse bool option, which does not work as expected.
      
      PiperOrigin-RevId: 248805505
      
      * Class to compute similarity between aggregated descriptors.
      
      PiperOrigin-RevId: 249102986
      
      * Script to perform retrieval and compute metrics.
      
      PiperOrigin-RevId: 249104011
      
      * feature_aggregation_similarity library in DELF init
      
      * D2R instructions / README update
      
      * Small edit to README
      
      * Internal change.
      
      PiperOrigin-RevId: 249113531
      
      * Instructions to reproduce D2R paper results, and small edits to config files.
      
      PiperOrigin-RevId: 249159850
      9062d200
  14. 16 May, 2019 2 commits
  15. 14 May, 2019 2 commits
    • derekjchow's avatar
      Move config to the right place. (#6782) · aecf5d02
      derekjchow authored
      aecf5d02
    • André Araujo's avatar
      DELF feature aggregation extraction: VLAD, ASMK, ASMK* (#6780) · a27bad9b
      André Araujo authored
      * Initial feature aggregation code for Detect-to-Retrieve paper.
      
      PiperOrigin-RevId: 246043144
      
      * Add support for ASMK/ASMK*/R-ASMK/R-ASMK*.
      
      PiperOrigin-RevId: 247337028
      
      * Add DatumProto uint32 field, and limit datum_io to uint32 and float32/float64 types.
      
      Also, introduce DatumPairProto, to be used for ASMK variants. Functions to read/write in this new format are added and tested.
      
      PiperOrigin-RevId: 247515205
      
      * Add batching option to feature aggregation extraction.
      
      PiperOrigin-RevId: 247614627
      
      * Script to perform local feature aggregation, with associated configs.
      
      Also small edits to the aggregation extractor, for better handling of input features / avoiding OOM.
      
      PiperOrigin-RevId: 248150750
      
      * Tests to check that aggregation using regions with no local features works.
      
      PiperOrigin-RevId: 248153275
      
      * Include new library/proto for aggregation
      
      * Merged commit includes the following changes:
      
      PiperOrigin-RevId: 248176511
      
      * Merged commit includes the following changes:
      248194572  by Andre Araujo:
      
          Change tf.tensor_scatter_nd_add --> tf.compat.v1.tensor_scatter_add to make it compatible with TF 1.X.
      
      --
      
      PiperOrigin-RevId: 248194572
      a27bad9b
  16. 08 May, 2019 1 commit
    • Menglong Zhu's avatar
      Merged commit includes the following changes: (#6726) · 58856e2b
      Menglong Zhu authored
      246873701  by menglong:
      
          Missing __init__.py under meta_architectures/
      
      --
      246857392  by menglong:
      
          Standardize proto namespace: lstm_object_detection.protos
      
      --
      246625127  by menglong:
      
          Internal changes.
      
      --
      246596481  by menglong:
      
          Add License
      
      --
      246580605  by menglong:
      
          Internal changes
      
      --
      246344626  by menglong:
      
          Open source interleaved mobilenet v2 model.
      
      --
      244893883  by menglong:
      
          Introduce multi_input_decoder for interleaved model.
      
      --
      244461016  by menglong:
      
          Add pre-bottleneck operation to lstm cells to support interleaved model.
      
      --
      244052176  by menglong:
      
          Update README
      
      --
      244020495  by menglong:
      
          Add test to rnn_decoder.
      
      --
      243704250  by menglong:
      
          Duplicate assignment.
      
      --
      243091836  by menglong:
      
          Move LSTMSSD meta arch into separate folder
      
      --
      242900337  by menglong:
      
          Modified mobilenet definition for LSTM-SSD
      
      --
      242773195  by menglong:
      
          Release GroupedConvLSTMCell implementation: https://arxiv.org/abs/1903.10172
      
      --
      242574736  by menglong:
      
          Introduce module for quantizated training.
      
      --
      242544306  by menglong:
      
          lstm_ssd_meta_arch updates, added test
          rename:
          - LSTMMetaArch to LSTMSSDMetaArch
          - LSTMFeatureExtractor to LSTMSSDFeatureExtractor
      
      --
      241986236  by menglong:
      
          Move lstm quantization utils to 3rd party.
      
      --
      225922488  by yinxiao:
      
          Training pipeline fixes.
      
      --
      224839137  by yinxiao:
      
          Issue fix for lstm object detecion sample config.
      
      --
      224246947  by menglong:
      
          Fix logging module import
      
      --
      
      PiperOrigin-RevId: 246873701
      58856e2b
  17. 06 May, 2019 1 commit
  18. 02 May, 2019 1 commit
    • Krishnan Srinivasan's avatar
      update mujoco and tensorflow 1.14/2.0 api calls in efficient-hrl (#6525) · 17ecf9db
      Krishnan Srinivasan authored
      * updated mujoco api calls to data, and tensorflow call to CriticalSection (moved from contrib.framework)
      
      * changed efficient-hrl envs so self.physics points to correct PyMj object based on mujoco_py version
      
      * corrected mujoco version number checkm in AntEnv.physics and PointEnv.physics properties
      
      * fix AntEnv.set_xy by reverting to using self.physics
      17ecf9db
  19. 01 May, 2019 1 commit
    • André Araujo's avatar
      Small edit to DELF README (#6704) · 62ed862e
      André Araujo authored
      * Internal change.
      
      PiperOrigin-RevId: 185564155
      
      * Fix small bug when reading DELF features when file is empty.
      A new test is added that catches this bug.
      
      PiperOrigin-RevId: 213839503
      
      * Refactors DELF example code to expose a function to create a DELF feature extractor.
      
      PiperOrigin-RevId: 241492615
      
      * Merged commit includes the following changes:
      244073180  by Andre Araujo:
      
          Internal change
      
      --
      243646498  by Andre Araujo:
      
          Detect and save bounding boxes for a list of images.
          A new proto is added, along with auxiliary read/write functions, and tests.
      
      --
      
      PiperOrigin-RevId: 244073180
      
      * Merged commit includes the following changes:
      244872693  by Andre Araujo:
      
          Small change.
      
      --
      244871213  by Andre Araujo:
      
          Small edit to DELF extractor function.
      
      --
      244790715  by Andre Araujo:
      
          internal automated change
      
      --
      244773146  by Andre Araujo:
      
          Add option to visualize extracted boxes, and change slightly model input.
      
      --
      244288992  by Andre Araujo:
      
          Script to cluster DELF features using K-means.
      
      --
      244275164  by Andre Araujo:
      
          Scripts to extract DELF features and boxes from Revisited Oxford/Paris datasets, used in Detect-to-Retrieve paper.
      
      --
      
      PiperOrigin-RevId: 244872693
      
      * Merged commit includes the following changes:
      
      PiperOrigin-RevId: 244876167
      
      * Updates instructions to include new released models, instructions for DELF and detector extraction, etc.
      
      * Small fix
      
      * Merged commit includes the following changes:
      244909336  by Andre Araujo:
      
          Update to DELF config example to use latest model.
      
      --
      
      PiperOrigin-RevId: 244909336
      
      * Revert "Merged commit includes the following changes:"
      
      This reverts commit f41be9f159330f62846d5d8b3d3d5d55cd1874f4.
      
      * Update delf_config_example model
      
      * Small edits to detection instructions
      
      * Specifying GLB location in README
      
      * Mention D2R code release in progress
      62ed862e
  20. 29 Apr, 2019 1 commit
  21. 26 Apr, 2019 1 commit
    • pkulzc's avatar
      Add VisualWakeWords Dataset to Slim dataset_factory (#6661) · 67c403fc
      pkulzc authored
      * Merged commit includes the following changes:
      244869387  by Sergio Guadarrama:
      
          This CL adds script/code to generate Visual WakeWords Dataset annotation files and TF records starting from COCO dataset.
      
      --
      244866660  by Sergio Guadarrama:
      
          Add VisualWakeWords Dataset to Slim dataset_factory to train MobileNets on it.
      
      --
      244836000  by Sergio Guadarrama:
      
          n/a
      
      --
      244104396  by Sergio Guadarrama:
      
          Add an option whether to enable / disable image cropping in inception_preprocessing.
      
      --
      242040128  by Sergio Guadarrama:
      
          Internal change
      
      241793677  by Sergio Guadarrama:
      
          Internal change
      
      241073081  by Sergio Guadarrama:
      
          Internal change
      
      240131189  by Sergio Guadarrama:
      
          Internal change
      
      PiperOrigin-RevId: 244869387
      
      * Merged commit includes the following changes:
      245431876  by Sergio Guadarrama:
      
          Internal cleanup
      
      --
      
      PiperOrigin-RevId: 245431876
      
      * Merged commit in...
      67c403fc
  22. 24 Apr, 2019 2 commits
    • André Araujo's avatar
      Latest updates from DELF code. (#6651) · 07678af8
      André Araujo authored
      * Internal change.
      
      PiperOrigin-RevId: 185564155
      
      * Fix small bug when reading DELF features when file is empty.
      A new test is added that catches this bug.
      
      PiperOrigin-RevId: 213839503
      
      * Refactors DELF example code to expose a function to create a DELF feature extractor.
      
      PiperOrigin-RevId: 241492615
      
      * Merged commit includes the following changes:
      244073180  by Andre Araujo:
      
          Internal change
      
      --
      243646498  by Andre Araujo:
      
          Detect and save bounding boxes for a list of images.
          A new proto is added, along with auxiliary read/write functions, and tests.
      
      --
      
      PiperOrigin-RevId: 244073180
      
      * Merged commit includes the following changes:
      244872693  by Andre Araujo:
      
          Small change.
      
      --
      244871213  by Andre Araujo:
      
          Small edit to DELF extractor function.
      
      --
      244790715  by Andre Araujo:
      
          internal automated change
      
      --
      244773146  by Andre Araujo:
      
          Add option to visualize extracted boxes, and change slightly model input.
      
      --
      244288992  by Andre Araujo:
      
          Script to cluster DELF features using K-means.
      
      --
      244275164  by Andre Araujo:
      
          Scripts to extract DELF features and boxes from Revisited Oxford/Paris datasets, used in Detect-to-Retrieve paper.
      
      --
      
      PiperOrigin-RevId: 244872693
      
      * Merged commit includes the following changes:
      
      PiperOrigin-RevId: 244876167
      
      * Updates instructions to include new released models, instructions for DELF and detector extraction, etc.
      
      * Small fix
      
      * Merged commit includes the following changes:
      244909336  by Andre Araujo:
      
          Update to DELF config example to use latest model.
      
      --
      
      PiperOrigin-RevId: 244909336
      
      * Revert "Merged commit includes the following changes:"
      
      This reverts commit f41be9f159330f62846d5d8b3d3d5d55cd1874f4.
      
      * Update delf_config_example model
      
      * Small edits to detection instructions
      
      * Specifying GLB location in README
      07678af8
    • Yukun Zhu's avatar
      Update colab demo (#6652) · 0d76b69f
      Yukun Zhu authored
      * Update colab demo
      
      * fix typo
      0d76b69f
  23. 23 Apr, 2019 1 commit
  24. 22 Apr, 2019 1 commit
  25. 20 Apr, 2019 1 commit
  26. 18 Apr, 2019 1 commit
    • André Araujo's avatar
      Recent DELF updates, including initial Detect-to-Retrieve code (#6599) · 61793e67
      André Araujo authored
      * Internal change.
      
      PiperOrigin-RevId: 185564155
      
      * Fix small bug when reading DELF features when file is empty.
      A new test is added that catches this bug.
      
      PiperOrigin-RevId: 213839503
      
      * Refactors DELF example code to expose a function to create a DELF feature extractor.
      
      PiperOrigin-RevId: 241492615
      
      * Merged commit includes the following changes:
      244073180  by Andre Araujo:
      
          Internal change
      
      --
      243646498  by Andre Araujo:
      
          Detect and save bounding boxes for a list of images.
          A new proto is added, along with auxiliary read/write functions, and tests.
      
      --
      
      PiperOrigin-RevId: 244073180
      61793e67
  27. 17 Apr, 2019 3 commits
  28. 08 Apr, 2019 1 commit
  29. 29 Mar, 2019 1 commit
  30. 26 Mar, 2019 1 commit