diff --git a/README.md b/README.md
index ac593fb4..34255fe7 100644
--- a/README.md
+++ b/README.md
@@ -66,7 +66,7 @@ Supported algorithms:
<details open>
<summary>Text Detection</summary>
-- [x] [DBNet](configs/textdet/dbnet/README.md) (AAAI'2020)
+- [x] [DBNet](configs/textdet/dbnet/README.md) (AAAI'2020) / [DBNet++](configs/textdet/dbnetpp/README.md) (TPAMI'2022)
- [x] [Mask R-CNN](configs/textdet/maskrcnn/README.md) (ICCV'2017)
- [x] [PANet](configs/textdet/panet/README.md) (ICCV'2019)
- [x] [PSENet](configs/textdet/psenet/README.md) (CVPR'2019)
diff --git a/README_zh-CN.md b/README_zh-CN.md
index b7070578..a4c58ad0 100644
--- a/README_zh-CN.md
+++ b/README_zh-CN.md
@@ -66,7 +66,7 @@ MMOCR 是基于 PyTorch 和 mmdetection 的开源工具箱，专注于文本检
<details open>
<summary>文字检测</summary>
-- [x] [DBNet](configs/textdet/dbnet/README.md) (AAAI'2020)
+- [x] [DBNet](configs/textdet/dbnet/README.md) (AAAI'2020) / [DBNet++](configs/textdet/dbnetpp/README.md) (TPAMI'2022)
- [x] [Mask R-CNN](configs/textdet/maskrcnn/README.md) (ICCV'2017)
- [x] [PANet](configs/textdet/panet/README.md) (ICCV'2019)
- [x] [PSENet](configs/textdet/psenet/README.md) (CVPR'2019)
diff --git a/configs/_base_/det_models/dbnetpp_r50dcnv2_fpnc.py b/configs/_base_/det_models/dbnetpp_r50dcnv2_fpnc.py
new file mode 100644
index 00000000..f8eaf2ff
--- /dev/null
+++ b/configs/_base_/det_models/dbnetpp_r50dcnv2_fpnc.py
@@ -0,0 +1,28 @@
+model = dict(
+ type='DBNet',
+ backbone=dict(
+ type='mmdet.ResNet',
+ depth=50,
+ num_stages=4,
+ out_indices=(0, 1, 2, 3),
+ frozen_stages=-1,
+ norm_cfg=dict(type='BN', requires_grad=True),
+ norm_eval=False,
+ style='pytorch',
+ dcn=dict(type='DCNv2', deform_groups=1, fallback_on_stride=False),
+ init_cfg=dict(type='Pretrained', checkpoint='torchvision://resnet50'),
+ stage_with_dcn=(False, True, True, True)),
+ neck=dict(
+ type='FPNC',
+ in_channels=[256, 512, 1024, 2048],
+ lateral_channels=256,
+ asf_cfg=dict(attention_type='ScaleChannelSpatial')),
+ bbox_head=dict(
+ type='DBHead',
+ in_channels=256,
+ loss=dict(type='DBLoss', alpha=5.0, beta=10.0, bbce_loss=True),
+ postprocessor=dict(
+ type='DBPostprocessor', text_repr_type='quad',
+ epsilon_ratio=0.002)),
+ train_cfg=None,
+ test_cfg=None)
diff --git a/configs/textdet/dbnet/metafile.yml b/configs/textdet/dbnet/metafile.yml
index 597fe42e..c6abdbca 100644
--- a/configs/textdet/dbnet/metafile.yml
+++ b/configs/textdet/dbnet/metafile.yml
@@ -5,7 +5,7 @@ Collections:
Training Techniques:
- SGD with Momentum
- Weight Decay
- Training Resources: 8x GeForce GTX 1080 Ti
+ Training Resources: 1x GeForce GTX 1080 Ti
Architecture:
- ResNet
- FPNC
diff --git a/configs/textdet/dbnetpp/README.md b/configs/textdet/dbnetpp/README.md
new file mode 100644
index 00000000..a3e08c3b
--- /dev/null
+++ b/configs/textdet/dbnetpp/README.md
@@ -0,0 +1,33 @@
+# DBNetpp
+
+> [Real-Time Scene Text Detection with Differentiable Binarization and Adaptive Scale Fusion](https://arxiv.org/abs/2202.10304)
+
+<!-- [ALGORITHM] -->
+
+## Abstract
+
+Recently, segmentation-based scene text detection methods have drawn extensive attention in the scene text detection field, because of their superiority in detecting the text instances of arbitrary shapes and extreme aspect ratios, profiting from the pixel-level descriptions. However, the vast majority of the existing segmentation-based approaches are limited to their complex post-processing algorithms and the scale robustness of their segmentation models, where the post-processing algorithms are not only isolated to the model optimization but also time-consuming and the scale robustness is usually strengthened by fusing multi-scale feature maps directly. In this paper, we propose a Differentiable Binarization (DB) module that integrates the binarization process, one of the most important steps in the post-processing procedure, into a segmentation network. Optimized along with the proposed DB module, the segmentation network can produce more accurate results, which enhances the accuracy of text detection with a simple pipeline. Furthermore, an efficient Adaptive Scale Fusion (ASF) module is proposed to improve the scale robustness by fusing features of different scales adaptively. By incorporating the proposed DB and ASF with the segmentation network, our proposed scene text detector consistently achieves state-of-the-art results, in terms of both detection accuracy and speed, on five standard benchmarks.
+
+<div align=center>
+<img src="https://user-images.githubusercontent.com/45810070/166850828-f1e48c25-4a0f-429d-ae54-6997ed25c062.png"/>
+</div>
+
+## Results and models
+
+### ICDAR2015
+
+| Method | Pretrained Model | Training set | Test set | #epochs | Test size | Recall | Precision | Hmean | Download |
+| :---------------------------------------------------------------------------: | :------------------------------------------------------------------------------------------------------------------------: | :-------------: | :------------: | :-----: | :-------: | :----: | :-------: | :---: | :-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
+| [DBNetpp_r50dcn](/configs/textdet/dbnetpp/dbnetpp_r50dcnv2_fpnc_1200e_icdar2015.py) | [Synthtext](/configs/textdet/dbnetpp/dbnetpp_r50dcnv2_fpnc_100k_iter_synthtext.py) ([model](https://download.openmmlab.com/mmocr/textdet/dbnet/dbnetpp_r50dcnv2_fpnc_100k_iter_synthtext-20220502-db297554.pth) \| [log](https://download.openmmlab.com/mmocr/textdet/dbnet/dbnetpp_r50dcnv2_fpnc_100k_iter_synthtext-20220502-db297554.log.json))| ICDAR2015 Train | ICDAR2015 Test | 1200 | 1024 | 0.822 | 0.901 | 0.860 | [model](https://download.openmmlab.com/mmocr/textdet/dbnet/dbnetpp_r50dcnv2_fpnc_1200e_icdar2015-20220502-d7a76fff.pth) \| [log](https://download.openmmlab.com/mmocr/textdet/dbnet/dbnetpp_r50dcnv2_fpnc_1200e_icdar2015-20220502-d7a76fff.log.json) |
+
+## Citation
+
+```bibtex
+@article{liao2022real,
+ title={Real-Time Scene Text Detection with Differentiable Binarization and Adaptive Scale Fusion},
+ author={Liao, Minghui and Zou, Zhisheng and Wan, Zhaoyi and Yao, Cong and Bai, Xiang},
+ journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
+ year={2022},
+ publisher={IEEE}
+}
+```
diff --git a/configs/textdet/dbnetpp/dbnetpp_r50dcnv2_fpnc_100k_iter_synthtext.py b/configs/textdet/dbnetpp/dbnetpp_r50dcnv2_fpnc_100k_iter_synthtext.py
new file mode 100644
index 00000000..cab4e9f7
--- /dev/null
+++ b/configs/textdet/dbnetpp/dbnetpp_r50dcnv2_fpnc_100k_iter_synthtext.py
@@ -0,0 +1,33 @@
+_base_ = [
+ '../../_base_/default_runtime.py',
+ '../../_base_/schedules/schedule_sgd_100k_iters.py',
+ '../../_base_/det_models/dbnetpp_r50dcnv2_fpnc.py',
+ '../../_base_/det_datasets/synthtext.py',
+ '../../_base_/det_pipelines/dbnet_pipeline.py'
+]
+
+train_list = {{_base_.train_list}}
+test_list = {{_base_.test_list}}
+
+train_pipeline_r50dcnv2 = {{_base_.train_pipeline_r50dcnv2}}
+test_pipeline_4068_1024 = {{_base_.test_pipeline_4068_1024}}
+
+data = dict(
+ samples_per_gpu=16,
+ workers_per_gpu=8,
+ val_dataloader=dict(samples_per_gpu=1),
+ test_dataloader=dict(samples_per_gpu=1),
+ train=dict(
+ type='UniformConcatDataset',
+ datasets=train_list,
+ pipeline=train_pipeline_r50dcnv2),
+ val=dict(
+ type='UniformConcatDataset',
+ datasets=test_list,
+ pipeline=test_pipeline_4068_1024),
+ test=dict(
+ type='UniformConcatDataset',
+ datasets=test_list,
+ pipeline=test_pipeline_4068_1024))
+
+evaluation = dict(interval=200000, metric='hmean-iou') # do not evaluate
diff --git a/configs/textdet/dbnetpp/dbnetpp_r50dcnv2_fpnc_1200e_icdar2015.py b/configs/textdet/dbnetpp/dbnetpp_r50dcnv2_fpnc_1200e_icdar2015.py
new file mode 100644
index 00000000..bc6ab78c
--- /dev/null
+++ b/configs/textdet/dbnetpp/dbnetpp_r50dcnv2_fpnc_1200e_icdar2015.py
@@ -0,0 +1,39 @@
+_base_ = [
+ '../../_base_/default_runtime.py',
+ '../../_base_/schedules/schedule_sgd_1200e.py',
+ '../../_base_/det_models/dbnetpp_r50dcnv2_fpnc.py',
+ '../../_base_/det_datasets/icdar2015.py',
+ '../../_base_/det_pipelines/dbnet_pipeline.py'
+]
+
+train_list = {{_base_.train_list}}
+test_list = {{_base_.test_list}}
+
+train_pipeline_r50dcnv2 = {{_base_.train_pipeline_r50dcnv2}}
+test_pipeline_4068_1024 = {{_base_.test_pipeline_4068_1024}}
+
+load_from = 'checkpoints/textdet/dbnetpp/res50dcnv2_synthtext.pth'
+
+data = dict(
+ samples_per_gpu=32,
+ workers_per_gpu=8,
+ val_dataloader=dict(samples_per_gpu=1),
+ test_dataloader=dict(samples_per_gpu=1),
+ train=dict(
+ type='UniformConcatDataset',
+ datasets=train_list,
+ pipeline=train_pipeline_r50dcnv2),
+ val=dict(
+ type='UniformConcatDataset',
+ datasets=test_list,
+ pipeline=test_pipeline_4068_1024),
+ test=dict(
+ type='UniformConcatDataset',
+ datasets=test_list,
+ pipeline=test_pipeline_4068_1024))
+
+evaluation = dict(
+ interval=100,
+ metric='hmean-iou',
+ save_best='0_hmean-iou:hmean',
+ rule='greater')
diff --git a/configs/textdet/dbnetpp/metafile.yml b/configs/textdet/dbnetpp/metafile.yml
new file mode 100644
index 00000000..b40571c1
--- /dev/null
+++ b/configs/textdet/dbnetpp/metafile.yml
@@ -0,0 +1,28 @@
+Collections:
+- Name: DBNetpp
+ Metadata:
+ Training Data: ICDAR2015
+ Training Techniques:
+ - SGD with Momentum
+ - Weight Decay
+ Training Resources: 1x Nvidia A100
+ Architecture:
+ - ResNet
+ - FPNC
+ Paper:
+ URL: https://arxiv.org/abs/2202.10304
+ Title: 'Real-Time Scene Text Detection with Differentiable Binarization and Adaptive Scale Fusion'
+ README: configs/textdet/dbnetpp/README.md
+
+Models:
+ - Name: dbnetpp_r50dcnv2_fpnc_1200e_icdar2015.py
+ In Collection: DBNetpp
+ Config: configs/textdet/dbnetpp/dbnetpp_r50dcnv2_fpnc_1200e_icdar2015.py
+ Metadata:
+ Training Data: ICDAR2015
+ Results:
+ - Task: Text Detection
+ Dataset: ICDAR2015
+ Metrics:
+ hmean-iou: 0.860
+ Weights: https://download.openmmlab.com/mmocr/textdet/dbnet/dbnetpp_r50dcnv2_fpnc_1200e_icdar2015-20220502-d7a76fff.pth
diff --git a/demo/README.md b/demo/README.md
index f5e5eb3d..2b492b9b 100644
--- a/demo/README.md
+++ b/demo/README.md
@@ -199,6 +199,7 @@ means that `batch_mode` and `print_result` are set to `True`)
| ------------- | :------------------------------------------------------------------------------------------------------------------------------------------------------: | :----------------------------: |
| DB_r18 | [link](https://mmocr.readthedocs.io/en/latest/textdet_models.html#real-time-scene-text-detection-with-differentiable-binarization) | :x: |
| DB_r50 | [link](https://mmocr.readthedocs.io/en/latest/textdet_models.html#real-time-scene-text-detection-with-differentiable-binarization) | :x: |
+| DBPP_r50 | [link](https://mmocr.readthedocs.io/en/latest/textdet_models.html#dbnetpp) | :x: |
| DRRG | [link](https://mmocr.readthedocs.io/en/latest/textdet_models.html#drrg) | :x: |
| FCE_IC15 | [link](https://mmocr.readthedocs.io/en/latest/textdet_models.html#fourier-contour-embedding-for-arbitrary-shaped-text-detection) | :x: |
| FCE_CTW_DCNv2 | [link](https://mmocr.readthedocs.io/en/latest/textdet_models.html#fourier-contour-embedding-for-arbitrary-shaped-text-detection) | :x: |
diff --git a/demo/README_zh-CN.md b/demo/README_zh-CN.md
index dd193b92..1b6545cb 100644
--- a/demo/README_zh-CN.md
+++ b/demo/README_zh-CN.md
@@ -196,6 +196,7 @@ mmocr 为了方便使用提供了预置的模型配置和对应的预训练权
| ------------- | :------------------------------------------------------------------------------------------------------------------------------------------------------: | :-------------------: |
| DB_r18 | [链接](https://mmocr.readthedocs.io/en/latest/textdet_models.html#real-time-scene-text-detection-with-differentiable-binarization) | :x: |
| DB_r50 | [链接](https://mmocr.readthedocs.io/en/latest/textdet_models.html#real-time-scene-text-detection-with-differentiable-binarization) | :x: |
+| DBPP_r50 | [链接](https://mmocr.readthedocs.io/en/latest/textdet_models.html#dbnetpp) | :x: |
| DRRG | [链接](https://mmocr.readthedocs.io/en/latest/textdet_models.html#drrg) | :x: |
| FCE_IC15 | [链接](https://mmocr.readthedocs.io/en/latest/textdet_models.html#fourier-contour-embedding-for-arbitrary-shaped-text-detection) | :x: |
| FCE_CTW_DCNv2 | [链接](https://mmocr.readthedocs.io/en/latest/textdet_models.html#fourier-contour-embedding-for-arbitrary-shaped-text-detection) | :x: |
diff --git a/mmocr/models/textdet/necks/fpn_cat.py b/mmocr/models/textdet/necks/fpn_cat.py
index 90d9d222..5acd2bcc 100644
--- a/mmocr/models/textdet/necks/fpn_cat.py
+++ b/mmocr/models/textdet/necks/fpn_cat.py
@@ -1,8 +1,9 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch
+import torch.nn as nn
import torch.nn.functional as F
from mmcv.cnn import ConvModule
-from mmcv.runner import BaseModule, ModuleList, auto_fp16
+from mmcv.runner import BaseModule, ModuleList, Sequential, auto_fp16
from mmocr.models.builder import NECKS
@@ -26,6 +27,8 @@ class FPNC(BaseModule):
bias_on_smooth (bool): Whether to use bias on smoothing layer.
bn_re_on_smooth (bool): Whether to use BatchNorm and ReLU on smoothing
layer.
+ asf_cfg (dict): Adaptive Scale Fusion module configs. The
+ attention_type can be 'ScaleChannelSpatial'.
conv_after_concat (bool): Whether to add a convolution layer after
the concatenation of predictions.
init_cfg (dict or list[dict], optional): Initialization configs.
@@ -39,8 +42,13 @@ class FPNC(BaseModule):
bn_re_on_lateral=False,
bias_on_smooth=False,
bn_re_on_smooth=False,
+ asf_cfg=None,
conv_after_concat=False,
- init_cfg=None):
+ init_cfg=[
+ dict(type='Kaiming', layer='Conv'),
+ dict(
+ type='Constant', layer='BatchNorm', val=1., bias=1e-4)
+ ]):
super().__init__(init_cfg=init_cfg)
assert isinstance(in_channels, list)
self.in_channels = in_channels
@@ -49,6 +57,7 @@ class FPNC(BaseModule):
self.num_ins = len(in_channels)
self.bn_re_on_lateral = bn_re_on_lateral
self.bn_re_on_smooth = bn_re_on_smooth
+ self.asf_cfg = asf_cfg
self.conv_after_concat = conv_after_concat
self.lateral_convs = ModuleList()
self.smooth_convs = ModuleList()
@@ -88,6 +97,24 @@ class FPNC(BaseModule):
self.lateral_convs.append(l_conv)
self.smooth_convs.append(smooth_conv)
+
+ if self.asf_cfg is not None:
+ self.asf_conv = ConvModule(
+ out_channels * self.num_outs,
+ out_channels * self.num_outs,
+ 3,
+ padding=1,
+ conv_cfg=None,
+ norm_cfg=None,
+ act_cfg=None,
+ inplace=False)
+ if self.asf_cfg['attention_type'] == 'ScaleChannelSpatial':
+ self.asf_attn = ScaleChannelSpatialAttention(
+ self.out_channels * self.num_outs,
+ (self.out_channels * self.num_outs) // 4, self.num_outs)
+ else:
+ raise NotImplementedError
+
if self.conv_after_concat:
norm_cfg = dict(type='BN')
act_cfg = dict(type='ReLU')
@@ -135,9 +162,110 @@ class FPNC(BaseModule):
for i, out in enumerate(outs):
outs[i] = F.interpolate(
outs[i], size=outs[0].shape[2:], mode='nearest')
+
out = torch.cat(outs, dim=1)
+ if self.asf_cfg is not None:
+ asf_feature = self.asf_conv(out)
+ attention = self.asf_attn(asf_feature)
+ enhanced_feature = []
+ for i, out in enumerate(outs):
+ enhanced_feature.append(attention[:, i:i + 1] * outs[i])
+ out = torch.cat(enhanced_feature, dim=1)
if self.conv_after_concat:
out = self.out_conv(out)
return out
+
+
+class ScaleChannelSpatialAttention(BaseModule):
+ """Spatial Attention module in Real-Time Scene Text Detection with
+ Differentiable Binarization and Adaptive Scale Fusion.
+
+ This was partially adapted from https://github.com/MhLiao/DB
+
+ Args:
+ in_channels (int): A numbers of input channels.
+ c_wise_channels (int): Number of channel-wise attention channels.
+ out_channels (int): Number of output channels.
+ init_cfg (dict or list[dict], optional): Initialization configs.
+ """
+
+ def __init__(self,
+ in_channels,
+ c_wise_channels,
+ out_channels,
+ init_cfg=[dict(type='Kaiming', layer='Conv', bias=0)]):
+ super().__init__(init_cfg=init_cfg)
+ self.avg_pool = nn.AdaptiveAvgPool2d(1)
+ # Channel Wise
+ self.channel_wise = Sequential(
+ ConvModule(
+ in_channels,
+ c_wise_channels,
+ 1,
+ bias=False,
+ conv_cfg=None,
+ norm_cfg=None,
+ act_cfg=dict(type='ReLU'),
+ inplace=False),
+ ConvModule(
+ c_wise_channels,
+ in_channels,
+ 1,
+ bias=False,
+ conv_cfg=None,
+ norm_cfg=None,
+ act_cfg=dict(type='Sigmoid'),
+ inplace=False))
+ # Spatial Wise
+ self.spatial_wise = Sequential(
+ ConvModule(
+ 1,
+ 1,
+ 3,
+ padding=1,
+ bias=False,
+ conv_cfg=None,
+ norm_cfg=None,
+ act_cfg=dict(type='ReLU'),
+ inplace=False),
+ ConvModule(
+ 1,
+ 1,
+ 1,
+ bias=False,
+ conv_cfg=None,
+ norm_cfg=None,
+ act_cfg=dict(type='Sigmoid'),
+ inplace=False))
+ # Attention Wise
+ self.attention_wise = ConvModule(
+ in_channels,
+ out_channels,
+ 1,
+ bias=False,
+ conv_cfg=None,
+ norm_cfg=None,
+ act_cfg=dict(type='Sigmoid'),
+ inplace=False)
+
+ @auto_fp16()
+ def forward(self, inputs):
+ """
+ Args:
+ inputs (Tensor): A concat FPN feature tensor that has the shape of
+ :math:`(N, C, H, W)`.
+
+ Returns:
+ Tensor: An attention map of shape :math:`(N, C_{out}, H, W)`
+ where :math:`C_{out}` is ``out_channels``.
+ """
+ out = self.avg_pool(inputs)
+ out = self.channel_wise(out)
+ out = out + inputs
+ inputs = torch.mean(out, dim=1, keepdim=True)
+ out = self.spatial_wise(inputs) + out
+ out = self.attention_wise(out)
+
+ return out
diff --git a/mmocr/models/textdet/postprocess/db_postprocessor.py b/mmocr/models/textdet/postprocess/db_postprocessor.py
index d9dbbeb2..f185b633 100644
--- a/mmocr/models/textdet/postprocess/db_postprocessor.py
+++ b/mmocr/models/textdet/postprocess/db_postprocessor.py
@@ -21,6 +21,7 @@ class DBPostprocessor(BasePostprocessor):
min_text_width (int): The minimum width of boundary polygon/box
predicted.
unclip_ratio (float): The unclip ratio for text regions dilation.
+ epsilon_ratio (float): The epsilon ratio for approximation accuracy.
max_candidates (int): The maximum candidate number.
"""
@@ -30,6 +31,7 @@ class DBPostprocessor(BasePostprocessor):
min_text_score=0.3,
min_text_width=5,
unclip_ratio=1.5,
+ epsilon_ratio=0.01,
max_candidates=3000,
**kwargs):
super().__init__(text_repr_type)
@@ -37,6 +39,7 @@ class DBPostprocessor(BasePostprocessor):
self.min_text_score = min_text_score
self.min_text_width = min_text_width
self.unclip_ratio = unclip_ratio
+ self.epsilon_ratio = epsilon_ratio
self.max_candidates = max_candidates
def __call__(self, preds):
@@ -62,7 +65,7 @@ class DBPostprocessor(BasePostprocessor):
for i, poly in enumerate(contours):
if i > self.max_candidates:
break
- epsilon = 0.01 * cv2.arcLength(poly, True)
+ epsilon = self.epsilon_ratio * cv2.arcLength(poly, True)
approx = cv2.approxPolyDP(poly, epsilon, True)
points = approx.reshape((-1, 2))
if points.shape[0] < 4:
diff --git a/mmocr/utils/ocr.py b/mmocr/utils/ocr.py
index 9304b20c..c0d6d8df 100755
--- a/mmocr/utils/ocr.py
+++ b/mmocr/utils/ocr.py
@@ -201,6 +201,13 @@ class MMOCR:
'dbnet/'
'dbnet_r50dcnv2_fpnc_sbn_1200e_icdar2015_20211025-9fe3b590.pth'
},
+ 'DBPP_r50': {
+ 'config':
+ 'dbnetpp/dbnetpp_r50dcnv2_fpnc_1200e_icdar2015.py',
+ 'ckpt':
+ 'dbnet/'
+ 'dbnetpp_r50dcnv2_fpnc_1200e_icdar2015-20220502-d7a76fff.pth'
+ },
'DRRG': {
'config':
'drrg/drrg_r50_fpn_unet_1200e_ctw1500.py',
diff --git a/tests/test_models/test_textdet_neck.py b/tests/test_models/test_textdet_neck.py
index 7bee9d7e..4410a4d1 100644
--- a/tests/test_models/test_textdet_neck.py
+++ b/tests/test_models/test_textdet_neck.py
@@ -9,14 +9,20 @@ def test_fpnc():
in_channels = [64, 128, 256, 512]
size = [112, 56, 28, 14]
+ asf_cfgs = [
+ None,
+ dict(attention_type='ScaleChannelSpatial'),
+ ]
for flag in [False, True]:
- fpnc = FPNC(
- in_channels=in_channels,
- bias_on_lateral=flag,
- bn_re_on_lateral=flag,
- bias_on_smooth=flag,
- bn_re_on_smooth=flag,
- conv_after_concat=flag)
+ for asf_cfg in asf_cfgs:
+ fpnc = FPNC(
+ in_channels=in_channels,
+ bias_on_lateral=flag,
+ bn_re_on_lateral=flag,
+ bias_on_smooth=flag,
+ bn_re_on_smooth=flag,
+ asf_cfg=asf_cfg,
+ conv_after_concat=flag)
fpnc.init_weights()
inputs = []
for i in range(4):