mmsegmentation/mmseg/ops/__init__.py
robin Han 0c04f52c42
Onnx upsample (#100)
* add customized Upsample which can convert to ONNX

* support multiply decode head for hrnet

* support size for Upsample
2020-09-03 19:59:13 +08:00

6 lines
212 B
Python

from .encoding import Encoding
from .separable_conv_module import DepthwiseSeparableConvModule
from .wrappers import Upsample, resize
__all__ = ['Upsample', 'resize', 'DepthwiseSeparableConvModule', 'Encoding']