support multi-batch for faster-rcnn & mask-rcnn (#998)

ce5ab172 · wangguanzhong · GitHub · 40437a31 · ce5ab172 · ce5ab172
隐藏空白更改
内联并排

Showing with 8 addition and 0 deletion

configs/faster_reader.yml configs/faster_reader.yml +4 -0

configs/mask_reader.yml configs/mask_reader.yml +4 -0

未找到文件。
--- a/configs/faster_reader.yml
+++ b/configs/faster_reader.yml
@@ -24,6 +24,10 @@ TrainReader:
  - !Permute
    to_bgr: false
    channel_first: true
+  batch_transforms:
+  - !PadBatch
+    pad_to_stride: -1.
+    use_padded_im_info: false
  batch_size: 1
  shuffle: true
  worker_num: 2

--- a/configs/mask_reader.yml
+++ b/configs/mask_reader.yml
@@ -25,6 +25,10 @@ TrainReader:
  - !Permute
    to_bgr: false
    channel_first: true
+  batch_transforms:
+  - !PadBatch
+    pad_to_stride: -1.
+    use_padded_im_info: false
  batch_size: 1
  shuffle: true
  worker_num: 2