未验证 提交 2a2644d6 编写于 作者: W Wenyu 提交者: GitHub

Fix args for infer (#3840)

上级 8a9e268d
...@@ -39,6 +39,6 @@ TestReader: ...@@ -39,6 +39,6 @@ TestReader:
- Permute: {} - Permute: {}
batch_transforms: batch_transforms:
- PadBatch: {pad_to_stride: 32} - PadBatch: {pad_to_stride: 32}
- Gt2SparseRCNNTarget: {train: false} - Gt2SparseRCNNTarget: {}
batch_size: 1 batch_size: 1
shuffle: false shuffle: false
...@@ -767,7 +767,6 @@ class Gt2SparseRCNNTarget(BaseOperator): ...@@ -767,7 +767,6 @@ class Gt2SparseRCNNTarget(BaseOperator):
sample["scale_factor_wh"] = np.array( sample["scale_factor_wh"] = np.array(
[sample["scale_factor"][1], sample["scale_factor"][0]], [sample["scale_factor"][1], sample["scale_factor"][0]],
dtype=np.float32) dtype=np.float32)
sample.pop("scale_factor")
else: else:
sample["scale_factor_wh"] = np.array( sample["scale_factor_wh"] = np.array(
[1.0, 1.0], dtype=np.float32) [1.0, 1.0], dtype=np.float32)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册