From 1483720cc0c45606364f6ef3c978ac60e57330d6 Mon Sep 17 00:00:00 2001 From: Ziyi Wu Date: Mon, 19 Apr 2021 23:52:42 +0800 Subject: [PATCH] add BaseSegmentor import to segmentors/__init__.py (#495) --- mmseg/models/segmentors/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mmseg/models/segmentors/__init__.py b/mmseg/models/segmentors/__init__.py index 3f600ecb9f..dca2f09405 100644 --- a/mmseg/models/segmentors/__init__.py +++ b/mmseg/models/segmentors/__init__.py @@ -1,4 +1,5 @@ +from .base import BaseSegmentor from .cascade_encoder_decoder import CascadeEncoderDecoder from .encoder_decoder import EncoderDecoder -__all__ = ['EncoderDecoder', 'CascadeEncoderDecoder'] +__all__ = ['BaseSegmentor', 'EncoderDecoder', 'CascadeEncoderDecoder']