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 81b5ef72874e805513e87231e525a2e6539ad03c..bacd53b75421bce4f2e95ba96ec56704ae4bd4be 100644 --- a/ppdet/data/transform/batch_operators.py +++ b/ppdet/data/transform/batch_operators.py @@ -767,7 +767,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)