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

Fix args for infer (#3840)

上级 8a9e268d
......@@ -39,6 +39,6 @@ TestReader:
- Permute: {}
batch_transforms:
- PadBatch: {pad_to_stride: 32}
- Gt2SparseRCNNTarget: {train: false}
- Gt2SparseRCNNTarget: {}
batch_size: 1
shuffle: false
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册