-
Notifications
You must be signed in to change notification settings - Fork 343
/
psenet_r50_tt.py
58 lines (58 loc) · 1.16 KB
/
psenet_r50_tt.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
model = dict(
type='PSENet',
backbone=dict(
type='resnet50',
pretrained=True
),
neck=dict(
type='FPN',
in_channels=(256, 512, 1024, 2048),
out_channels=128
),
detection_head=dict(
type='PSENet_Head',
in_channels=1024,
hidden_dim=256,
num_classes=7,
loss_text=dict(
type='DiceLoss',
loss_weight=0.7
),
loss_kernel=dict(
type='DiceLoss',
loss_weight=0.3
)
)
)
data = dict(
batch_size=16,
train=dict(
type='PSENET_TT',
split='train',
is_transform=True,
img_size=736,
short_size=736,
kernel_num=7,
min_scale=0.7,
read_type='cv2'
),
test=dict(
type='PSENET_TT',
split='test',
short_size=736,
read_type='cv2'
)
)
train_cfg = dict(
lr=1e-3,
schedule=(200, 400,),
epoch=600,
optimizer='SGD'
)
test_cfg = dict(
min_score=0.87,
min_area=16,
kernel_num=7,
bbox_type='poly',
result_path='outputs/submit_tt/'
)