MMOCR / configs /_base_ /det_models /panet_r50_fpem_ffm.py
Robin Chiu
Add the configs
a92ae91
model = dict(
type='PANet',
pretrained='torchvision://resnet50',
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=True,
style='caffe'),
neck=dict(type='FPEM_FFM', in_channels=[256, 512, 1024, 2048]),
bbox_head=dict(
type='PANHead',
in_channels=[128, 128, 128, 128],
out_channels=6,
loss=dict(type='PANLoss', speedup_bbox_thr=32),
postprocessor=dict(type='PANPostprocessor', text_repr_type='poly')),
train_cfg=None,
test_cfg=None)