From 62a01f7e30acfe739f008aa0870348ecfb6fcdff Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Tue, 26 Apr 2022 17:29:38 +0800 Subject: [PATCH] fix --- configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml | 4 ++-- configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml | 2 +- ppocr/modeling/heads/det_db_head.py | 8 ++++---- 3 files changed, 7 insertions(+), 7 deletions(-) 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 6a838fd8..95da6d24 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 a9e1d3a6..0e04cd9b 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 2d94555a..5539d773 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()), -- GitLab