diff --git a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml index 6a838fd8d03c83754a07294406fe4ba575761643..95da6d24c2ce947f7bc422e082338b1e26f0efce 100644 --- a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml +++ b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml @@ -32,7 +32,7 @@ NetWorks: model_name: large disable_se: true Neck: - name: CAPAN + name: CAFPN out_channels: 96 shortcut: True Head: @@ -48,7 +48,7 @@ NetWorks: model_name: large disable_se: true Neck: - name: CAPAN + name: CAFPN out_channels: 96 shortcut: True Head: diff --git a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml index a9e1d3a6a256c42e7b203d83d3c84ed7e48263d0..0e04cd9b06c4908c7fa3b41aa6a09fdd41a1f56c 100644 --- a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml +++ b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml @@ -28,7 +28,7 @@ Architecture: model_name: large disable_se: True Neck: - name: CAPAN + name: CAFPN out_channels: 96 shortcut: True Head: diff --git a/ppocr/modeling/heads/det_db_head.py b/ppocr/modeling/heads/det_db_head.py index 2d94555aeb9ef61c25a014fd4c20f468292ee90c..5539d7737ed8993166a90d0bec600ce6ead3c2a5 100644 --- a/ppocr/modeling/heads/det_db_head.py +++ b/ppocr/modeling/heads/det_db_head.py @@ -37,8 +37,8 @@ class Head(nn.Layer): self.conv1 = nn.Conv2D( in_channels=in_channels, out_channels=in_channels // 4, - kernel_size=kernel_size[0], - padding=int(kernel_size[0] // 2), + kernel_size=kernel_list[0], + padding=int(kernel_list[0] // 2), weight_attr=ParamAttr(), bias_attr=False) self.conv_bn1 = nn.BatchNorm( @@ -51,7 +51,7 @@ class Head(nn.Layer): self.conv2 = nn.Conv2DTranspose( in_channels=in_channels // 4, out_channels=in_channels // 4, - kernel_size=kernel_size[1], + kernel_size=kernel_list[1], stride=2, weight_attr=ParamAttr( initializer=paddle.nn.initializer.KaimingUniform()), @@ -66,7 +66,7 @@ class Head(nn.Layer): self.conv3 = nn.Conv2DTranspose( in_channels=in_channels // 4, out_channels=1, - kernel_size=kernel_size[2], + kernel_size=kernel_list[2], stride=2, weight_attr=ParamAttr( initializer=paddle.nn.initializer.KaimingUniform()),