forked from WenmuZhou/PAN.pytorch
/
config.json
79 lines (79 loc) · 1.8 KB
/
config.json
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
{
"name": "PAN",
"data_loader": {
"type": "ImageDataset",
"args": {
"alphabet": "alphabet.npy",
"dataset": {
"train_data_path": [
[
"E:\\zj\\dataset\\icdar2015\\train\\train.txt"
]
],
"train_data_ratio": [
1.0
],
"val_data_path": [
],
"input_size": 640,
"img_channel": 3,
"shrink_ratio": 0.5
},
"loader": {
"validation_split": 0.1,
"train_batch_size": 8,
"val_batch_size": 1,
"shuffle": true,
"pin_memory": false,
"num_workers": 6
}
}
},
"arch": {
"type": "PANModel",
"args": {
"backbone": "resnet18",
"fpem_repeat": 2,
"pretrained": true
}
},
"loss": {
"type": "PANLoss",
"args": {
"alpha": 0.5,
"beta": 0.25,
"delta_agg": 0.5,
"delta_dis": 3,
"ohem_ratio": 3
}
},
"optimizer": {
"type": "Adam",
"args": {
"lr": 0.001,
"weight_decay": 0,
"amsgrad": true
}
},
"lr_scheduler": {
"type": "StepLR",
"args": {
"step_size": 200,
"gamma": 0.1
}
},
"trainer": {
"seed": 2,
"gpus": [
0
],
"epochs": 600,
"display_interval": 10,
"resume": {
"restart_training": true,
"checkpoint": ""
},
"output_dir": "output",
"tensorboard": true
}
}