Browse Source

Added input check fix to random rotation

tags/v1.2.0-rc1
Eric 5 years ago
parent
commit
47f5e2867d
2 changed files with 6 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/sync_wait_node.cc
  2. +5
    -0
      mindspore/dataset/vision/c_transforms.py

+ 1
- 1
mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/sync_wait_node.cc View File

@@ -46,7 +46,7 @@ Status SyncWaitNode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_o
// Right now barrier should only take num_rows_per_buffer = 1
// The reason for this is because having it otherwise can lead to blocking issues
// See barrier_op.h for more details
int32_t rows_per_buffer = 1;
const int32_t rows_per_buffer = 1;
auto op = std::make_shared<BarrierOp>(rows_per_buffer, connector_que_size_, condition_name_, callback_);
op->set_total_repeats(GetTotalRepeats());
op->set_num_repeats_per_epoch(GetNumRepeatsPerEpoch());


+ 5
- 0
mindspore/dataset/vision/c_transforms.py View File

@@ -1114,6 +1114,11 @@ class RandomRotation(ImageTensorOperation):

@check_random_rotation
def __init__(self, degrees, resample=Inter.NEAREST, expand=False, center=None, fill_value=0):
if isinstance(degrees, numbers.Number):
degrees = degrees % 360
if isinstance(degrees, (list, tuple)):
degrees = [degrees[0] % 360, degrees[1] % 360]

self.degrees = degrees
self.resample = resample
self.expand = expand


Loading…
Cancel
Save