未验证 提交 63ead1e4 编写于 作者: W Wenyu 提交者: GitHub

Fix args for infer (#3839)

上级 d6c8947f
...@@ -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
...@@ -895,7 +895,6 @@ class Gt2SparseRCNNTarget(BaseOperator): ...@@ -895,7 +895,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.
先完成此消息的编辑!
想要评论请 注册