forked from looput/PSENet-Tensorflow
-
Notifications
You must be signed in to change notification settings - Fork 0
/
configuration.py
64 lines (51 loc) · 1.16 KB
/
configuration.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
59
60
61
62
63
64
#! /usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright © 2018 loop Huazhong University of Science and Technology
# Config file for net and train
TRAIN_CONFIG={
'profile':False,
'train_image_shape':(640,640),
'data_format':'NHWC',
'use_rotation':False,
'm':0.5,
'n':2,
'OHM':True,
'weight_decay':5e-4,
'batch_size':8,
'log_dir':'Logs',
'lr_config':{
'lr_boundaries':[500,800],
'lr_values':[1e-3,1e-4,1e-5]
},
'deformable':False,
# dataset name
'data_name':'icdar2015',
'data_config':{
'read_num_p':2,
'para_num_p':2,
'pro_num_p':4,
'buffer_size':100,
'prefetch':50
},
'rate':[0.5],
'min_size':640,
'ran_scale':[0.5, 1.0, 2.0, 3.0]
}
TEST_CONFIG ={
'id': '0',
'ckpt': '',
'log_dir': 'Logs',
# TODO your path here
'test_dir': '/workspace/datasets/icdar2015/ch4_test_images',
'n':len(TRAIN_CONFIG['rate'])+1,
'threshold_kernel': 0.6,
'threshold': 0.55,
'aver_score': 0.89,
'image_size': {
'w':1280, # 2240
'h':768, # 1280
'fixed_size':False,
'scale':2
}
}