rename to

kaiko-ai · Oct 31, 2024 · 3446b0a · 3446b0a
1 parent 10ca27b
commit 3446b0a
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/docs/reference/vision/models/networks.md b/docs/reference/vision/models/networks.md
@@ -9,5 +9,5 @@ Reference information for the vision model `Networks` API.
 ::: eva.vision.models.networks.decoders.segmentation.ConvDecoderMS
 ::: eva.vision.models.networks.decoders.segmentation.LinearDecoder
 ::: eva.vision.models.networks.decoders.segmentation.SingleLinearDecoder
-::: eva.vision.models.networks.decoders.segmentation.ConvDecoderWithImagePrior
+::: eva.vision.models.networks.decoders.segmentation.ConvDecoderWithImage
 
diff --git a/src/eva/vision/models/networks/decoders/segmentation/__init__.py b/src/eva/vision/models/networks/decoders/segmentation/__init__.py
@@ -5,15 +5,15 @@
 from eva.vision.models.networks.decoders.segmentation.semantic import (
     ConvDecoder1x1,
     ConvDecoderMS,
-    ConvDecoderWithImagePrior,
+    ConvDecoderWithImage,
     SingleLinearDecoder,
 )
 
 __all__ = [
     "ConvDecoder1x1",
     "ConvDecoderMS",
     "SingleLinearDecoder",
-    "ConvDecoderWithImagePrior",
+    "ConvDecoderWithImage",
     "Decoder2D",
     "LinearDecoder",
 ]
diff --git a/src/eva/vision/models/networks/decoders/segmentation/semantic/__init__.py b/src/eva/vision/models/networks/decoders/segmentation/semantic/__init__.py
@@ -5,8 +5,8 @@
     ConvDecoderMS,
     SingleLinearDecoder,
 )
-from eva.vision.models.networks.decoders.segmentation.semantic.with_image_prior import (
-    ConvDecoderWithImagePrior,
+from eva.vision.models.networks.decoders.segmentation.semantic.with_image import (
+    ConvDecoderWithImage,
 )
 
-__all__ = ["ConvDecoder1x1", "ConvDecoderMS", "SingleLinearDecoder", "ConvDecoderWithImagePrior"]
+__all__ = ["ConvDecoder1x1", "ConvDecoderMS", "SingleLinearDecoder", "ConvDecoderWithImage"]
diff --git a/...segmentation/semantic/with_image_prior.py → ...oders/segmentation/semantic/with_image.py b/...segmentation/semantic/with_image_prior.py → ...oders/segmentation/semantic/with_image.py
@@ -11,7 +11,7 @@
 from eva.vision.models.networks.decoders.segmentation.typings import DecoderInputs
 
 
-class ConvDecoderWithImagePrior(decoder2d.Decoder2D):
+class ConvDecoderWithImage(decoder2d.Decoder2D):
     """A convolutional that in addition to encoded features, also takes the input image as input.
 
     In a first stage, the input features are upsampled and passed through a convolutional layer,