Browse Source

adjust folder of image transform method

tags/v1.8.0
ZhidanLiu 3 years ago
parent
commit
46cc2d2d13
10 changed files with 25 additions and 8 deletions
  1. +1
    -1
      examples/natural_robustness/natural_robustness_example.py
  2. +1
    -1
      examples/natural_robustness/serving/server/perturbation/servable_config.py
  3. +16
    -0
      mindarmour/natural_robustness/transform/__init__.py
  4. +0
    -0
      mindarmour/natural_robustness/transform/image/__init__.py
  5. +1
    -1
      mindarmour/natural_robustness/transform/image/blur.py
  6. +1
    -1
      mindarmour/natural_robustness/transform/image/corruption.py
  7. +1
    -1
      mindarmour/natural_robustness/transform/image/luminance.py
  8. +0
    -0
      mindarmour/natural_robustness/transform/image/natural_perturb.py
  9. +1
    -1
      mindarmour/natural_robustness/transform/image/transformation.py
  10. +3
    -2
      tests/ut/python/natural_robustness/test_natural_robustness.py

+ 1
- 1
examples/natural_robustness/natural_robustness_example.py View File

@@ -17,7 +17,7 @@
import numpy as np
import cv2

from mindarmour.natural_robustness.image import Translate, Curve, Perspective, Scale, Shear, Rotate, SaltAndPepperNoise, \
from mindarmour.natural_robustness.transform.image import Translate, Curve, Perspective, Scale, Shear, Rotate, SaltAndPepperNoise, \
NaturalNoise, GaussianNoise, UniformNoise, MotionBlur, GaussianBlur, GradientBlur, Contrast, GradientLuminance




+ 1
- 1
examples/natural_robustness/serving/server/perturbation/servable_config.py View File

@@ -21,7 +21,7 @@ import cv2
import numpy as np
from PIL import Image
from mindspore_serving.server import register
from mindarmour.natural_robustness.image import Contrast, GaussianBlur, SaltAndPepperNoise, Scale, Shear, \
from mindarmour.natural_robustness.transform.image import Contrast, GaussianBlur, SaltAndPepperNoise, Scale, Shear, \
Translate, Rotate, MotionBlur, GradientBlur, GradientLuminance, NaturalNoise, Curve, Perspective




+ 16
- 0
mindarmour/natural_robustness/transform/__init__.py View File

@@ -0,0 +1,16 @@
# Copyright 2022 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""
Image Transform Method for Natural Robustness.
"""

mindarmour/natural_robustness/image/__init__.py → mindarmour/natural_robustness/transform/image/__init__.py View File


mindarmour/natural_robustness/image/blur.py → mindarmour/natural_robustness/transform/image/blur.py View File

@@ -18,7 +18,7 @@ Image Blur
import numpy as np
import cv2

from mindarmour.natural_robustness.image.natural_perturb import _NaturalPerturb
from mindarmour.natural_robustness.transform.image.natural_perturb import _NaturalPerturb
from mindarmour.utils._check_param import check_param_multi_types, check_int_positive, check_param_type
from mindarmour.utils.logger import LogUtil


mindarmour/natural_robustness/image/corruption.py → mindarmour/natural_robustness/transform/image/corruption.py View File

@@ -18,7 +18,7 @@ import math
import numpy as np
import cv2

from mindarmour.natural_robustness.image.natural_perturb import _NaturalPerturb
from mindarmour.natural_robustness.transform.image.natural_perturb import _NaturalPerturb
from mindarmour.utils._check_param import check_param_multi_types, check_param_type
from mindarmour.utils.logger import LogUtil


mindarmour/natural_robustness/image/luminance.py → mindarmour/natural_robustness/transform/image/luminance.py View File

@@ -18,7 +18,7 @@ import math
import numpy as np
import cv2

from mindarmour.natural_robustness.image.natural_perturb import _NaturalPerturb
from mindarmour.natural_robustness.transform.image.natural_perturb import _NaturalPerturb
from mindarmour.utils._check_param import check_param_multi_types, check_param_in_range, check_param_type, \
check_value_non_negative
from mindarmour.utils.logger import LogUtil

mindarmour/natural_robustness/image/natural_perturb.py → mindarmour/natural_robustness/transform/image/natural_perturb.py View File


mindarmour/natural_robustness/image/transformation.py → mindarmour/natural_robustness/transform/image/transformation.py View File

@@ -18,7 +18,7 @@ import math
import numpy as np
import cv2

from mindarmour.natural_robustness.image.natural_perturb import _NaturalPerturb
from mindarmour.natural_robustness.transform.image.natural_perturb import _NaturalPerturb
from mindarmour.utils._check_param import check_param_multi_types, check_param_type, check_value_non_negative
from mindarmour.utils.logger import LogUtil


+ 3
- 2
tests/ut/python/natural_robustness/test_natural_robustness.py View File

@@ -18,8 +18,9 @@ import pytest
import numpy as np
from mindspore import context

from mindarmour.natural_robustness.image import Translate, Curve, Perspective, Scale, Shear, Rotate, SaltAndPepperNoise, \
NaturalNoise, GaussianNoise, UniformNoise, MotionBlur, GaussianBlur, GradientBlur, Contrast, GradientLuminance
from mindarmour.natural_robustness.transform.image import Translate, Curve, Perspective, Scale, Shear, Rotate, \
SaltAndPepperNoise, NaturalNoise, GaussianNoise, UniformNoise, MotionBlur, GaussianBlur, GradientBlur, Contrast,\
GradientLuminance


@pytest.mark.level0


Loading…
Cancel
Save