Added annotations

peiva-git · Dec 16, 2023 · 75fb88c · 75fb88c
1 parent 0f2b94c
commit 75fb88c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/rancrop_model_variants/mean_aggregation/pp_liteseg_rancrops_pad_after_inference.py b/rancrop_model_variants/mean_aggregation/pp_liteseg_rancrops_pad_after_inference.py
@@ -6,8 +6,10 @@
 
 import paddle as pp
 from paddleseg.models import PPLiteSeg
+from paddleseg.cvlibs import manager
 
 
+@manager.MODELS.add_component
 class PPLiteSegRandomCrops(PPLiteSeg):
     """
     This class represents an extension of the PPLiteSeg model.

diff --git a/...p_model_variants/softmax_plus_mean_aggregation/pp_liteseg_rancrops_pad_after_inference.py b/...p_model_variants/softmax_plus_mean_aggregation/pp_liteseg_rancrops_pad_after_inference.py
@@ -6,8 +6,10 @@
 
 import paddle as pp
 from paddleseg.models import PPLiteSeg
+from paddleseg.cvlibs import manager
 
 
+@manager.MODELS.add_component
 class PPLiteSegRandomCrops(PPLiteSeg):
     """
     This class represents an extension of the PPLiteSeg model.

diff --git a/..._model_variants/softmax_plus_mean_aggregation/pp_liteseg_rancrops_pad_before_inference.py b/..._model_variants/softmax_plus_mean_aggregation/pp_liteseg_rancrops_pad_before_inference.py
@@ -6,8 +6,10 @@
 
 import paddle as pp
 from paddleseg.models import PPLiteSeg
+from paddleseg.cvlibs import manager
 
 
+@manager.MODELS.add_component
 class PPLiteSegRandomCrops(PPLiteSeg):
     """
     This class represents an extension of the PPLiteSeg model.