!7679 Modify calibration parameters

Merge pull request !7679 from shenwei41/question
pull/7679/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit e10d55165a

@ -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_);

Loading…
Cancel
Save