diff --git a/configs/sparse_rcnn/_base_/sparse_rcnn_reader.yml b/configs/sparse_rcnn/_base_/sparse_rcnn_reader.yml index f4226473d78168c407337ee156b25ab7b7462b67..248ca39b741b0607ab053a911a954ef2719964c2 100644 --- a/configs/sparse_rcnn/_base_/sparse_rcnn_reader.yml +++ b/configs/sparse_rcnn/_base_/sparse_rcnn_reader.yml @@ -39,6 +39,6 @@ TestReader: - Permute: {} batch_transforms: - PadBatch: {pad_to_stride: 32} - - Gt2SparseRCNNTarget: {train: false} + - Gt2SparseRCNNTarget: {} batch_size: 1 shuffle: false diff --git a/ppdet/data/transform/batch_operators.py b/ppdet/data/transform/batch_operators.py index e72635898a1d55311737976e502cdff2ef3a6fe0..6bb4bf0e5896bc7fc4dfb0444f253cb46aa1f1ef 100644 --- a/ppdet/data/transform/batch_operators.py +++ b/ppdet/data/transform/batch_operators.py @@ -895,7 +895,6 @@ class Gt2SparseRCNNTarget(BaseOperator): sample["scale_factor_wh"] = np.array( [sample["scale_factor"][1], sample["scale_factor"][0]], dtype=np.float32) - sample.pop("scale_factor") else: sample["scale_factor_wh"] = np.array( [1.0, 1.0], dtype=np.float32)