forked from Newcomer520/tf-dagmm
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
51 lines (46 loc) · 2.01 KB
/
config.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
FILTERS = [32, 64, 96, 128, 196, 256]
FILTERS_128 = [32, 64, 80, 96, 128, 196, 256]
FILTERS_D6 = [32, 64, 128, 196, 256, 320]
FILTERS_D6_SMALL = [32, 64, 96, 128, 196, 256]
FILTERS_D5 = [32, 64, 96, 128, 160]
FILTERS_D5_SMALL = [16, 32, 64, 96, 128]
FILTERS_D4 = [32, 64, 96, 128]
FILTERS_SMALL = [8, 16, 32, 48, 64, 96]
REGIONS_PIN = [(67, 10, 112, 69), (277, 10, 322, 69)]
FILTERS_PIN = [32, 64, 96, 128, 196]
OBJECT_J0602 = {
'main': {'width': 128, 'height': 128, 'region': 'all', 'filters': FILTERS_128, 'scope': 'main', 'reuse': False},
}
OBJECT_J0602_PIN = {
'pin1': {'width': 64, 'height': 64, 'region': (67, 10, 112, 69), 'filters': FILTERS_SMALL, 'scope': 'up', 'reuse': False},
'pin2': {'width': 64, 'height': 64, 'region': (277, 10, 322, 69), 'filters': FILTERS_SMALL, 'scope': 'up', 'reuse': True},
'pin3': {'width': 64, 'height': 64, 'region': (105, 275, 285, 320), 'filters': FILTERS_SMALL, 'scope': 'down', 'reuse': False},
'main': OBJECT_J0602['main']
}
OBJECT_J0601 = {
'main': {'width': 320, 'height': 128, 'region': 'all', 'filters': FILTERS_D6, 'scope': 'main', 'reuse': False}
}
OBJECT_J0601_SPLIT = {
'main': {'width': 320, 'height': 64, 'region': [(0, 0, 694, 63), (0, 216, 694, 279)], 'filters': FILTERS_D6_SMALL, 'scope': 'main', 'reuse': False}
}
OBJECT_12567 = {
'main': {'width': 64, 'height': 64, 'region': 'all', 'filters': FILTERS_D5, 'scope': 'main', 'reuse': False}
}
OBJECT_12567_SMALL = {
'main': {'width': 64, 'height': 64, 'region': 'all', 'filters': FILTERS_D5_SMALL, 'scope': 'main', 'reuse': False}
}
OBJECT_32 = {
'main': {'width': 32, 'height': 32, 'region': 'all', 'filters': FILTERS_D4, 'scope': 'main', 'reuse': False}
}
def get_region(region_name):
regions = {
'default': OBJECT_J0602,
'J0602': OBJECT_J0602,
'J0602P': OBJECT_J0602_PIN,
'J0601': OBJECT_J0601,
'J0601_S': OBJECT_J0601_SPLIT,
'12567': OBJECT_12567,
'12567_S': OBJECT_12567_SMALL,
'32': OBJECT_32,
}
return regions[region_name]