未验证 提交 cb20bef6 编写于 作者: C cnn 提交者: GitHub

fix stride data type (#2645)

上级 143e22d8
...@@ -7,7 +7,7 @@ TrainReader: ...@@ -7,7 +7,7 @@ TrainReader:
- NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]} - NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]}
- Permute: {} - Permute: {}
batch_transforms: batch_transforms:
- PadBatch: {pad_to_stride: -1., pad_gt: true} - PadBatch: {pad_to_stride: -1, pad_gt: true}
batch_size: 1 batch_size: 1
shuffle: true shuffle: true
drop_last: true drop_last: true
...@@ -20,7 +20,7 @@ EvalReader: ...@@ -20,7 +20,7 @@ EvalReader:
- NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]} - NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]}
- Permute: {} - Permute: {}
batch_transforms: batch_transforms:
- PadBatch: {pad_to_stride: -1., pad_gt: false} - PadBatch: {pad_to_stride: -1, pad_gt: false}
batch_size: 1 batch_size: 1
shuffle: false shuffle: false
drop_last: false drop_last: false
...@@ -34,7 +34,11 @@ TestReader: ...@@ -34,7 +34,11 @@ TestReader:
- NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]} - NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]}
- Permute: {} - Permute: {}
batch_transforms: batch_transforms:
<<<<<<< HEAD
- PadBatch: {pad_to_stride: -1., pad_gt: false} - PadBatch: {pad_to_stride: -1., pad_gt: false}
=======
- PadBatch: {pad_to_stride: -1}
>>>>>>> 0fc9950... fix stride data type
batch_size: 1 batch_size: 1
shuffle: false shuffle: false
drop_last: false drop_last: false
...@@ -7,7 +7,7 @@ TrainReader: ...@@ -7,7 +7,7 @@ TrainReader:
- NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]} - NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]}
- Permute: {} - Permute: {}
batch_transforms: batch_transforms:
- PadBatch: {pad_to_stride: -1., pad_gt: true} - PadBatch: {pad_to_stride: -1, pad_gt: true}
batch_size: 1 batch_size: 1
shuffle: true shuffle: true
drop_last: true drop_last: true
...@@ -20,7 +20,11 @@ EvalReader: ...@@ -20,7 +20,11 @@ EvalReader:
- NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]} - NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]}
- Permute: {} - Permute: {}
batch_transforms: batch_transforms:
<<<<<<< HEAD
- PadBatch: {pad_to_stride: -1., pad_gt: false} - PadBatch: {pad_to_stride: -1., pad_gt: false}
=======
- PadBatch: {pad_to_stride: -1}
>>>>>>> 0fc9950... fix stride data type
batch_size: 1 batch_size: 1
shuffle: false shuffle: false
drop_last: false drop_last: false
...@@ -34,7 +38,11 @@ TestReader: ...@@ -34,7 +38,11 @@ TestReader:
- NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]} - NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]}
- Permute: {} - Permute: {}
batch_transforms: batch_transforms:
<<<<<<< HEAD
- PadBatch: {pad_to_stride: -1., pad_gt: false} - PadBatch: {pad_to_stride: -1., pad_gt: false}
=======
- PadBatch: {pad_to_stride: -1}
>>>>>>> 0fc9950... fix stride data type
batch_size: 1 batch_size: 1
shuffle: false shuffle: false
drop_last: false drop_last: false
......
...@@ -26,7 +26,7 @@ TrainReader: ...@@ -26,7 +26,7 @@ TrainReader:
channel_first: true channel_first: true
batch_transforms: batch_transforms:
- !PadBatch - !PadBatch
pad_to_stride: -1. pad_to_stride: -1
use_padded_im_info: false use_padded_im_info: false
batch_size: 1 batch_size: 1
shuffle: true shuffle: true
......
...@@ -27,7 +27,7 @@ TrainReader: ...@@ -27,7 +27,7 @@ TrainReader:
channel_first: true channel_first: true
batch_transforms: batch_transforms:
- !PadBatch - !PadBatch
pad_to_stride: -1. pad_to_stride: -1
use_padded_im_info: false use_padded_im_info: false
batch_size: 1 batch_size: 1
shuffle: true shuffle: true
......
...@@ -27,7 +27,7 @@ TrainReader: ...@@ -27,7 +27,7 @@ TrainReader:
channel_first: true channel_first: true
batch_transforms: batch_transforms:
- !PadBatch - !PadBatch
pad_to_stride: -1. pad_to_stride: -1
use_padded_im_info: false use_padded_im_info: false
batch_size: 1 batch_size: 1
shuffle: true shuffle: true
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册