* add customized Upsample which can convert to ONNX * support multiply decode head for hrnet * support size for Upsample
6 lines
212 B
Python
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']
|