未验证 提交 3a497c5f 编写于 作者: W wangguanzhong 提交者: GitHub

add image_shape in testreader for topdown models (#3543)

上级 999aa0d2
...@@ -128,6 +128,8 @@ EvalReader: ...@@ -128,6 +128,8 @@ EvalReader:
drop_empty: false drop_empty: false
TestReader: TestReader:
inputs_def:
image_shape: [3, *train_height, *train_width]
sample_transforms: sample_transforms:
- Decode: {} - Decode: {}
- TopDownEvalAffine: - TopDownEvalAffine:
......
...@@ -132,6 +132,8 @@ EvalReader: ...@@ -132,6 +132,8 @@ EvalReader:
drop_empty: false drop_empty: false
TestReader: TestReader:
inputs_def:
image_shape: [3, *train_height, *train_width]
sample_transforms: sample_transforms:
- Decode: {} - Decode: {}
- TopDownEvalAffine: - TopDownEvalAffine:
......
...@@ -128,6 +128,8 @@ EvalReader: ...@@ -128,6 +128,8 @@ EvalReader:
drop_empty: false drop_empty: false
TestReader: TestReader:
inputs_def:
image_shape: [3, *train_height, *train_width]
sample_transforms: sample_transforms:
- Decode: {} - Decode: {}
- TopDownEvalAffine: - TopDownEvalAffine:
......
...@@ -127,6 +127,8 @@ EvalReader: ...@@ -127,6 +127,8 @@ EvalReader:
batch_size: 16 batch_size: 16
TestReader: TestReader:
inputs_def:
image_shape: [3, *train_height, *train_width]
sample_transforms: sample_transforms:
- Decode: {} - Decode: {}
- TopDownEvalAffine: - TopDownEvalAffine:
......
...@@ -118,6 +118,8 @@ EvalReader: ...@@ -118,6 +118,8 @@ EvalReader:
batch_size: 16 batch_size: 16
TestReader: TestReader:
inputs_def:
image_shape: [3, *train_height, *train_width]
sample_transforms: sample_transforms:
- Decode: {} - Decode: {}
- TopDownEvalAffine: - TopDownEvalAffine:
......
...@@ -128,6 +128,8 @@ EvalReader: ...@@ -128,6 +128,8 @@ EvalReader:
batch_size: 16 batch_size: 16
TestReader: TestReader:
inputs_def:
image_shape: [3, *train_height, *train_width]
sample_transforms: sample_transforms:
- Decode: {} - Decode: {}
- TopDownEvalAffine: - TopDownEvalAffine:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册