diff --git a/segmentors/__init__.py b/segmentors/__init__.py
index 3db4ed44..d6281f8d 100644
--- a/segmentors/__init__.py
+++ b/segmentors/__init__.py
@@ -1,5 +1,5 @@
-from .upernet import UPerNet, SiamDiffUPerNet, SiamConcUPerNet, MTUPerNet
+from .upernet import UPerNet, SiamDiffUPerNet, SiamConcUPerNet, MTUPerNet, RegUPerNet, RegMTUPerNet
 from .ltae import LTAE2d
 from .unet import UNet, SiamDiffUNet, SiamConcUNet
 
-all = ['UPerNet', 'SiamDiffUPerNet', 'SiamConcUPerNet', 'MTUPerNet', 'UNet', 'SiamDiffUNet', 'SiamConcatUNet']
\ No newline at end of file
+all = ['UPerNet', 'SiamDiffUPerNet', 'SiamConcUPerNet', 'MTUPerNet', 'RegUPerNet', 'RegMTUPerNet', 'UNet', 'SiamDiffUNet', 'SiamConcatUNet']