1. 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
  2. 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
  3. 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
  4. 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
  5. 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
  6. 03 Dec, 2018 1 commit
  7. 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
  8. 24 Oct, 2018 8 commits