Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Modify calibration parameters

This commit is contained in:
shenwei41 2020-10-23 14:59:50 +08:00
parent 7f209bdf12
commit 6f8e94abd0
1 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
mindspore/ccsrc/minddata/dataset/api/vision.cc
View File

@ -427,6 +427,12 @@ bool CropOperation::ValidateParams() {
return false;
}
}
for (int j = 0; j < coordinates_.size(); ++j) {
if (coordinates_[j] < 0) {
MS_LOG(ERROR) << "Crop: invalid coordinates, coordinates must be greater than 0, got: " << coordinates_[j];
return false;
}
}
return true;
}
@ -437,7 +443,10 @@ std::shared_ptr<TensorOp> CropOperation::Build() {
y = coordinates_[1];
height = size_[0];
width = size_[1];
width = size_[0];
if (size_.size() == 2) {
width = size_[1];
}
std::shared_ptr<CropOp> tensor_op = std::make_shared<CropOp>(x, y, height, width);
return tensor_op;
@ -879,7 +888,13 @@ std::shared_ptr<TensorOp> RandomCropDecodeResizeOperation::Build() {
// RandomHorizontalFlipOperation
RandomHorizontalFlipOperation::RandomHorizontalFlipOperation(float probability) : probability_(probability) {}
bool RandomHorizontalFlipOperation::ValidateParams() { return true; }
bool RandomHorizontalFlipOperation::ValidateParams() {
if (probability_ < 0.0 || probability_ > 1.0) {
MS_LOG(ERROR) << "RandomHorizontalFlip: probability must be between 0.0 and 1.0.";
return false;
}
return true;
}
std::shared_ptr<TensorOp> RandomHorizontalFlipOperation::Build() {
std::shared_ptr<RandomHorizontalFlipOp> tensor_op = std::make_shared<RandomHorizontalFlipOp>(probability_);
@ -1024,7 +1039,13 @@ std::shared_ptr<TensorOp> RandomSolarizeOperation::Build() {
// RandomVerticalFlipOperation
RandomVerticalFlipOperation::RandomVerticalFlipOperation(float probability) : probability_(probability) {}
bool RandomVerticalFlipOperation::ValidateParams() { return true; }
bool RandomVerticalFlipOperation::ValidateParams() {
if (probability_ < 0.0 || probability_ > 1.0) {
MS_LOG(ERROR) << "RandomVerticalFlip: probability must be between 0.0 and 1.0.";
return false;
}
return true;
}
std::shared_ptr<TensorOp> RandomVerticalFlipOperation::Build() {
std::shared_ptr<RandomVerticalFlipOp> tensor_op = std::make_shared<RandomVerticalFlipOp>(probability_);