Add ResNet-UNet with ConvNext blocks.

PiperOrigin-RevId: 637065295
pyoung2778 · May 24, 2024 · 12353b7 · 12353b7
1 parent d93c7e9
commit 12353b7
Show file tree

Hide file tree

Showing 3 changed files with 669 additions and 1 deletion.
diff --git a/official/vision/modeling/backbones/__init__.py b/official/vision/modeling/backbones/__init__.py
@@ -14,13 +14,13 @@
 
 """Backbones package definition."""
 
-from official.projects.maskconver.modeling.resnet_unet import ResNetUNet
 from official.vision.modeling.backbones.efficientnet import EfficientNet
 from official.vision.modeling.backbones.mobiledet import MobileDet
 from official.vision.modeling.backbones.mobilenet import MobileNet
 from official.vision.modeling.backbones.resnet import ResNet
 from official.vision.modeling.backbones.resnet_3d import ResNet3D
 from official.vision.modeling.backbones.resnet_deeplab import DilatedResNet
+from official.vision.modeling.backbones.resnet_unet import ResNetUNet
 from official.vision.modeling.backbones.revnet import RevNet
 from official.vision.modeling.backbones.spinenet import SpineNet
 from official.vision.modeling.backbones.spinenet_mobile import SpineNetMobile