From ce5ab17279c96a540b3d381d4379a35612e60dfe Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Tue, 30 Jun 2020 19:18:07 +0800 Subject: [PATCH] support multi-batch for faster-rcnn & mask-rcnn (#998) --- configs/faster_reader.yml | 4 ++++ configs/mask_reader.yml | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/configs/faster_reader.yml b/configs/faster_reader.yml index 643b4721a..26835d4c6 100644 --- 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 diff --git a/configs/mask_reader.yml b/configs/mask_reader.yml index 50432ece6..31742f205 100644 --- 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 -- GitLab