custom_models->rs_models

README.md

@@ -159,7 +159,7 @@ PaddleRS是遥感科研院所、相关高校共同基于飞桨开发的遥感处
 ├── deploy               # 部署相关的文档和脚本
 ├── docs                 # 整个项目文档及图片
 ├── paddlers  
-│     ├── custom_models  # 自定义网络模型代码
+│     ├── rs_models      # 遥感专用网络模型代码
 │     ├── datasets       # 数据加载相关代码
 │     ├── models         # 套件网络模型代码
 │     ├── tasks          # 相关任务代码

docs/apis/model_zoo.md

@@ -2,7 +2,7 @@
 
 PaddleRS的基础模型库来自[PaddleClas](https://github.com/PaddlePaddle/PaddleClas/blob/release/2.3/docs/zh_CN/algorithm_introduction/ImageNet_models.md)、[PaddleSeg](https://github.com/PaddlePaddle/PaddleSeg/blob/release/2.4/docs/model_zoo_overview_cn.md)、[PaddleDetection](https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.3/README_cn.md#模型库)以及[PaddleGAN](https://github.com/PaddlePaddle/PaddleGAN/blob/develop/README_cn.md#模型库)，可以通过相关的链接进行查看。而在此之外，PaddleRS也针对遥感任务添加了一些特有的模型库，可用于遥感图像语义分割、遥感变化检测等。
 
-## 自定义模型库
+## 遥感专用模型库
 
 | 模型名称        | 用途     |
 | --------------- | -------- |
@@ -20,7 +20,7 @@ PaddleRS的基础模型库来自[PaddleClas](https://github.com/PaddlePaddle/Pad
 
 ## 如何导入
 
-模型均位于`paddlers/models`和`paddlers/custom_models`中，对于套件中的模型可以通过如下方法进行使用
+模型均位于`paddlers/models`和`paddlers/rs_models`中，对于套件中的模型可以通过如下方法进行使用
 
 ```python
 from paddlers.models import xxxx
@@ -29,5 +29,5 @@ from paddlers.models import xxxx
 而PaddleRS所特有的模型可以通过如下方法调用
 
 ```python
-from paddlers.custom_models import xxxx
+from paddlers.rs_models import xxxx
 ```

paddlers/custom_models/cd/changestar.py → paddlers/rs_models/cd/changestar.py

@@ -17,7 +17,7 @@ import paddle.nn as nn
 import paddle.nn.functional as F
 
 from paddlers.datasets.cd_dataset import MaskType
-from paddlers.custom_models.seg import FarSeg
+from paddlers.rs_models.seg import FarSeg
 from .layers import Conv3x3, Identity
 
 

paddlers/custom_models/cd/param_init.py → paddlers/rs_models/cd/param_init.py

@@ -67,7 +67,7 @@ class KaimingInitMixin:
 
     Examples:
 
-        from paddlers.custom_models.cd.models.param_init import KaimingInitMixin
+        from paddlers.rs_models.cd.models.param_init import KaimingInitMixin
 
         class CustomNet(nn.Layer, KaimingInitMixin):
             def __init__(self, num_channels, num_classes):

paddlers/tasks/change_detector.py

@@ -25,7 +25,7 @@ import paddle.nn.functional as F
 from paddle.static import InputSpec
 
 import paddlers
-import paddlers.custom_models.cd as cmcd
+import paddlers.rs_models.cd as cmcd
 import paddlers.utils.logging as logging
 import paddlers.models.ppseg as paddleseg
 from paddlers.transforms import arrange_transforms

paddlers/tasks/classifier.py

@@ -23,7 +23,7 @@ import paddle.nn.functional as F
 from paddle.static import InputSpec
 
 import paddlers.models.ppcls as paddleclas
-import paddlers.custom_models.cls as cmcls
+import paddlers.rs_models.cls as cmcls
 import paddlers
 from paddlers.transforms import arrange_transforms
 from paddlers.utils import get_single_card_bs, DisablePrint

paddlers/tasks/segmenter.py

@@ -24,7 +24,7 @@ import paddle.nn.functional as F
 from paddle.static import InputSpec
 
 import paddlers.models.ppseg as paddleseg
-import paddlers.custom_models.seg as cmseg
+import paddlers.rs_models.seg as cmseg
 import paddlers
 from paddlers.transforms import arrange_transforms
 from paddlers.utils import get_single_card_bs, DisablePrint

tests/rs_models/test_cd_models.py

@@ -65,7 +65,7 @@ class TestCDModel(TestModel):
 
 
 class TestBITModel(TestCDModel):
-    MODEL_CLASS = paddlers.custom_models.cd.BIT
+    MODEL_CLASS = paddlers.rs_models.cd.BIT
 
     def set_specs(self):
         base_spec = dict(in_channels=3, num_classes=2)
@@ -80,7 +80,7 @@ class TestBITModel(TestCDModel):
 
 
 class TestCDNetModel(TestCDModel):
-    MODEL_CLASS = paddlers.custom_models.cd.CDNet
+    MODEL_CLASS = paddlers.rs_models.cd.CDNet
     EF_MODE = 'Concat'
 
     def set_specs(self):
@@ -92,7 +92,7 @@ class TestCDNetModel(TestCDModel):
 
 
 class TestChangeStarModel(TestCDModel):
-    MODEL_CLASS = paddlers.custom_models.cd.ChangeStar
+    MODEL_CLASS = paddlers.rs_models.cd.ChangeStar
 
     def set_specs(self):
         self.specs = [
@@ -112,7 +112,7 @@ class TestChangeStarModel(TestCDModel):
 
 
 class TestDSAMNetModel(TestCDModel):
-    MODEL_CLASS = paddlers.custom_models.cd.DSAMNet
+    MODEL_CLASS = paddlers.rs_models.cd.DSAMNet
 
     def set_specs(self):
         base_spec = dict(in_channels=3, num_classes=2)
@@ -134,7 +134,7 @@ class TestDSAMNetModel(TestCDModel):
 
 
 class TestDSIFNModel(TestCDModel):
-    MODEL_CLASS = paddlers.custom_models.cd.DSIFN
+    MODEL_CLASS = paddlers.rs_models.cd.DSIFN
 
     def set_specs(self):
         self.specs = [
@@ -153,7 +153,7 @@ class TestDSIFNModel(TestCDModel):
 
 
 class TestFCEarlyFusionModel(TestCDModel):
-    MODEL_CLASS = paddlers.custom_models.cd.FCEarlyFusion
+    MODEL_CLASS = paddlers.rs_models.cd.FCEarlyFusion
     EF_MODE = 'Concat'
 
     def set_specs(self):
@@ -166,7 +166,7 @@ class TestFCEarlyFusionModel(TestCDModel):
 
 
 class TestFCSiamConcModel(TestCDModel):
-    MODEL_CLASS = paddlers.custom_models.cd.FCSiamConc
+    MODEL_CLASS = paddlers.rs_models.cd.FCSiamConc
 
     def set_specs(self):
         self.specs = [
@@ -178,7 +178,7 @@ class TestFCSiamConcModel(TestCDModel):
 
 
 class TestFCSiamDiffModel(TestCDModel):
-    MODEL_CLASS = paddlers.custom_models.cd.FCSiamDiff
+    MODEL_CLASS = paddlers.rs_models.cd.FCSiamDiff
 
     def set_specs(self):
         self.specs = [
@@ -190,7 +190,7 @@ class TestFCSiamDiffModel(TestCDModel):
 
 
 class TestSNUNetModel(TestCDModel):
-    MODEL_CLASS = paddlers.custom_models.cd.SNUNet
+    MODEL_CLASS = paddlers.rs_models.cd.SNUNet
 
     def set_specs(self):
         self.specs = [
@@ -202,7 +202,7 @@ class TestSNUNetModel(TestCDModel):
 
 
 class TestSTANetModel(TestCDModel):
-    MODEL_CLASS = paddlers.custom_models.cd.STANet
+    MODEL_CLASS = paddlers.rs_models.cd.STANet
 
     def set_specs(self):
         base_spec = dict(in_channels=3, num_classes=2)

tests/rs_models/test_seg_models.py

@@ -48,7 +48,7 @@ class TestSegModel(TestModel):
 
 
 class TestFarSegModel(TestSegModel):
-    MODEL_CLASS = paddlers.custom_models.seg.FarSeg
+    MODEL_CLASS = paddlers.rs_models.seg.FarSeg
 
     def set_specs(self):
         self.specs = [