add BaseSegmentor import to segmentors/__init__.py (#495)

open-mmlab · Apr 19, 2021 · 1483720 · 1483720
1 parent 8f37e55
commit 1483720
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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']