1. 03 Mar, 2020 1 commit
    • Liangzhe's avatar
      Merged commit includes the following changes: (#8235) · 4a0b3e4b
      Liangzhe authored
      
      
      298416930  by lzyuan:
      
          Explicitly mark base models' state inputs as 'raw_inputs/init_lstm_h_1' and 'raw_inputs_init_lstm_h_2' when pre_bottleneck=True.
      
      --
      298380851  by skligys:
      
          Fix LSTD LSTM cells to use fixed_quantize_op().
      
      --
      297662737  by Menglong Zhu:
      
          Explicitly replace "import tensorflow" with "tensorflow.compat.v1" for TF2.x migration
      
      --
      289667197  by lzyuan:
      
          Internal update.
      
      --
      288607438  by lzyuan:
      
          Enforce feature_extractor construction using arg keys.
      
      --
      
      PiperOrigin-RevId: 298416930
      Co-authored-by: default avatarMenglong Zhu <menglong@google.com>
      4a0b3e4b
  2. 08 Jan, 2020 1 commit
    • Skirmantas Kligys's avatar
      Update for tensorflow.contrib removal. (#8010) · 9302933b
      Skirmantas Kligys authored
      * Push conflicting earlier changes in files touched.
      
      PiperOrigin-RevId: 268414407
      PiperOrigin-RevId: 278006953
      PiperOrigin-RevId: 281567252
      PiperOrigin-RevId: 288363770
      
      * Removes tf.contrib usage and replaces them with direct imports for lstm object detection.
      
      PiperOrigin-RevId: 286065032
      
      * Internal changes.
      
      PiperOrigin-RevId: 288363770
      9302933b
  3. 06 Dec, 2019 1 commit
    • Skirmantas Kligys's avatar
      Push CLs 283653640, 284011539 (#7921) · 22e20f84
      Skirmantas Kligys authored
      * Manually quantize selected inputs before summing them up.
      
      PiperOrigin-RevId: 283653640
      
      * Fix messy variable paths in LSTD head.
      
      With a VariableScope passed as parameter into tf.variable_scope(),
      TF gets all confused and starts placing variables into scopes different
      from operations (variable_scope != name_scope).  That leads to very messy
      TF graphs.
      
      Before CL:
        FeatureExtractor/LSTM/conv_lstm_cell/bottleneck_0/depthwise_weights
        FeatureExtractor/LSTM/LSTM/conv_lstm_cell/bottleneck_0/separable_conv2d/depthwise
      
      After CL:
        FeatureExtractor/LSTM/conv_lstm_cell/bottleneck_0/depthwise_weights
        FeatureExtractor/LSTM/conv_lstm_cell/bottleneck_0/separable_conv2d/depthwise
      
      PiperOrigin-RevId: 284011539
      22e20f84
  4. 14 Nov, 2019 1 commit
  5. 27 Aug, 2019 1 commit
  6. 22 Aug, 2019 1 commit
    • Yongzhe Wang's avatar
      Resubmitting changes which have been reverted. (#7492) · 6252e588
      Yongzhe Wang authored
      * Merged commit includes the following changes:
      263863588  by yongzhe:
      
          Fix a bug that the SetExternalContext for EdgeTPU wasn't called when initializing LSTD client.
      
      --
      263370193  by yongzhe:
      
          Internal change.
      
      --
      
      PiperOrigin-RevId: 263863588
      
      * Revert changes in seq_dataset_builder_test.py
      
      * Remove stale code
      6252e588
  7. 19 Aug, 2019 2 commits
  8. 16 Aug, 2019 1 commit
  9. 14 Aug, 2019 1 commit
  10. 12 Aug, 2019 1 commit
    • Richard Brooks's avatar
      Lstm object detection improvements (#7379) · 49075e50
      Richard Brooks authored
      * Replace google3.pyglib modules with tf and absl
      
      This now matches train.py and provides more publicly available libraries.
      
      * Add example pipeline config for SSD Interleaved V2 Model.
      
      Compiled from model_builder_test.py and lstm_ssd_mobilenet_v1_imagenet.config,
      Removed data augmentation and tranfer learning (i.e. training from checkpoint) due to errors I was seeing when trying to run with it.
      
      * Remove unused tfrecord creation.
      
      This was also incorrectly specified, as the keys differed from the TFSequenceExample parser.
      
      * correct key specified in docstring
      
      * add tflite frozen graph exporter (cli and lib).
      
      * add tflite model exporter
      
      * add script to test the tflite model
      
      * add mode export documentation
      
      * correct docstring
      
      * rename export files to be unique across detection research work
      
      * correct number of channels for grayscale
      
      * add and correct copyright
      49075e50
  11. 02 Aug, 2019 1 commit
    • Yongzhe Wang's avatar
      Merged commit includes the following changes: (#7358) · 13e7c85d
      Yongzhe Wang authored
      261196859  by yongzhe:
      
          Integrate EdgeTPU API into the Mobile SSD tflite client.
      
          Build command with EdgeTPU enabled:
          bazel build mobile_ssd_tflite_client  --define enable_edgetpu=true
      
          Build command with EdgeTPU disabled:
          bazel build mobile_ssd_tflite_client
      
      --
      259096620  by Menglong Zhu:
      
          Remove unused proto imports.
      
      --
      
      PiperOrigin-RevId: 261196859
      13e7c85d
  12. 18 Jul, 2019 1 commit
    • Yongzhe Wang's avatar
      Merged commit includes the following changes: (#7249) · b7221961
      Yongzhe Wang authored
      * Merged commit includes the following changes:
      257930561  by yongzhe:
      
          Mobile LSTD TfLite Client.
      
      --
      257928126  by yongzhe:
      
          Mobile SSD Tflite client.
      
      --
      257921181  by menglong:
      
          Fix discrepancy between pre_bottleneck = {true, false}
      
      --
      257561213  by yongzhe:
      
          File utils.
      
      --
      257449226  by yongzhe:
      
          Mobile SSD Client.
      
      --
      257264654  by yongzhe:
      
          SSD utils.
      
      --
      257235648  by yongzhe:
      
          Proto bazel build rules.
      
      --
      256437262  by Menglong Zhu:
      
          Fix check for FusedBatchNorm op to only verify it as a prefix.
      
      --
      256283755  by yongzhe:
      
          Bazel build and copybara changes.
      
      --
      251947295  by yinxiao:
      
          Add missing interleaved option in checkpoint restore.
      
      --
      251513479  by yongzhe:
      
          Conversion utils.
      
      --
      248783193  by yongzhe:
      
          Branch protos needed for the lstd client.
      
      --
      248200507  by menglong:
      
          Fix proto namespace in example config
      
      --
      
      P...
      b7221961
  13. 16 Jul, 2019 1 commit
    • yongzhe2160's avatar
      Merged commit includes the following changes: (#7220) · 66d00a87
      yongzhe2160 authored
      * Merged commit includes the following changes:
      257930561  by yongzhe:
      
          Mobile LSTD TfLite Client.
      
      --
      257928126  by yongzhe:
      
          Mobile SSD Tflite client.
      
      --
      257921181  by menglong:
      
          Fix discrepancy between pre_bottleneck = {true, false}
      
      --
      257561213  by yongzhe:
      
          File utils.
      
      --
      257449226  by yongzhe:
      
          Mobile SSD Client.
      
      --
      257264654  by yongzhe:
      
          SSD utils.
      
      --
      257235648  by yongzhe:
      
          Proto bazel build rules.
      
      --
      256437262  by Menglong Zhu:
      
          Fix check for FusedBatchNorm op to only verify it as a prefix.
      
      --
      256283755  by yongzhe:
      
          Bazel build and copybara changes.
      
      --
      251947295  by yinxiao:
      
          Add missing interleaved option in checkpoint restore.
      
      --
      251513479  by yongzhe:
      
          Conversion utils.
      
      --
      248783193  by yongzhe:
      
          Branch protos needed for the lstd client.
      
      --
      248200507  by menglong:
      
          Fix proto namespace in example config
      
      --
      
      PiperOrigin-RevId: 257930561
      
      * Delete BUILD
      66d00a87
  14. 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
  15. 03 Dec, 2018 1 commit
  16. 26 Nov, 2018 3 commits
    • Menglong Zhu's avatar
      Fix dependency issues (#5815) · 2d9d7943
      Menglong Zhu authored
      * Add a second expected_loss_weights function using an alternative expectation calculation compared to previous. Integrate this op into ssd_meta_arch and losses builder. Affects files that use losses_builder.build to handle the returning of an additional element.
      
      PiperOrigin-RevId: 219190083
      
      * Fix dependency issues.
      
      PiperOrigin-RevId: 222888231
      
      * Update CODEOWNERS 
      
      Add @masonliuw and @yinxiaoli to /research/lstm_object_detection/
      2d9d7943
    • Menglong Zhu's avatar
      Fix dependency issues. · b915db4e
      Menglong Zhu authored
      PiperOrigin-RevId: 222888231
      b915db4e
    • Menglong Zhu's avatar
      Add a second expected_loss_weights function using an alternative expectation... · 8ab3e7a0
      Menglong Zhu authored
      Add a second expected_loss_weights function using an alternative expectation calculation compared to previous. Integrate this op into ssd_meta_arch and losses builder. Affects files that use losses_builder.build to handle the returning of an additional element.
      
      PiperOrigin-RevId: 219190083
      8ab3e7a0
  17. 24 Oct, 2018 8 commits