From b44f2bc41c41cd3da2b79400aaf260592e4cc4cb Mon Sep 17 00:00:00 2001
From: Ritu Yadav <40523539+RituYadav92@users.noreply.github.com>
Date: Fri, 20 Sep 2024 20:30:03 +0200
Subject: [PATCH] Update __init__.py

---
 segmentors/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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']