提交 4688bbab 编写于 作者: M Megvii Engine Team

refactor(functional): hide functional.vision and replace with functional.nn

GitOrigin-RevId: b6888c7f6578263fde5cadcb537c8c9a273b7ce8
上级 8796586b
......@@ -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
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册