Update __init__.py

VMarsocci · Sep 20, 2024 · b44f2bc · b44f2bc
1 parent 9129a49
commit b44f2bc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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']
+all = ['UPerNet', 'SiamDiffUPerNet', 'SiamConcUPerNet', 'MTUPerNet', 'RegUPerNet', 'RegMTUPerNet', 'UNet', 'SiamDiffUNet', 'SiamConcatUNet']