From 63ead1e4980738742c188078af24a2d889180dd7 Mon Sep 17 00:00:00 2001 From: Wenyu Date: Tue, 3 Aug 2021 13:10:00 +0800 Subject: [PATCH] Fix args for infer (#3839) --- configs/sparse_rcnn/_base_/sparse_rcnn_reader.yml | 2 +- ppdet/data/transform/batch_operators.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/configs/sparse_rcnn/_base_/sparse_rcnn_reader.yml b/configs/sparse_rcnn/_base_/sparse_rcnn_reader.yml index f4226473d..248ca39b7 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 e72635898..6bb4bf0e5 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) -- GitLab