diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py index 568df43f..db467791 100644 --- a/imperative/python/megengine/functional/nn.py +++ b/imperative/python/megengine/functional/nn.py @@ -76,7 +76,6 @@ __all__ = [ "relu", "relu6", "remap", - "resize", "sigmoid", "sliding_window", "sliding_window_transpose", @@ -1647,17 +1646,7 @@ def pad( return output -interpolate = deprecated_func("1.3", "megengine.functional.vision", "interpolate", True) -roi_pooling = deprecated_func("1.3", "megengine.functional.vision", "roi_pooling", True) -roi_align = deprecated_func("1.3", "megengine.functional.vision", "roi_align", True) -nms = deprecated_func("1.3", "megengine.functional.vision", "nms", True) -resize = deprecated_func("1.3", "megengine.functional.vision", "resize", True) -remap = deprecated_func("1.3", "megengine.functional.vision", "remap", True) -nvof = deprecated_func("1.3", "megengine.functional.vision", "nvof", True) -warp_affine = deprecated_func("1.3", "megengine.functional.vision", "warp_affine", True) -warp_perspective = deprecated_func( - "1.3", "megengine.functional.vision", "warp_perspective", True -) from .quantized import conv_bias_activation # isort:skip from .loss import * # isort:skip from .metric import * # isort:skip +from .vision import * # isort:skip diff --git a/imperative/python/megengine/functional/vision.py b/imperative/python/megengine/functional/vision.py index e79587f4..a0a2e197 100644 --- a/imperative/python/megengine/functional/vision.py +++ b/imperative/python/megengine/functional/vision.py @@ -19,6 +19,19 @@ from .elemwise import floor from .math import argsort from .tensor import broadcast_to, concat, expand_dims, reshape, transpose +__all__ = [ + "correlation", + "cvt_color", + "roi_pooling", + "roi_align", + "nms", + "remap", + "warp_affine", + "warp_perspective", + "interpolate", + "nvof", +] + def cvt_color(inp: Tensor, mode: str = ""): r"""Convert images from one format to another