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