/
config.py
97 lines (86 loc) · 2.88 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
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
# config.py
import os.path
# gets home dir cross platform
#HOME = os.path.expanduser("~")
#HOME = os.path.abspath(os.path.dirname(__file__)).split("/") this path
HOME = os.path.join(os.getcwd()) #../path
# for making bounding boxes pretty
COLORS = ((255, 0, 0, 128), (0, 255, 0, 128), (0, 0, 255, 128),
(0, 255, 255, 128), (255, 0, 255, 128), (255, 255, 0, 128))
#crcak (104, 117, 123)
# SSD300 CONFIGS
voc= {
'model':"resnet50",
'losstype':'Ciou',
'num_classes':21,
'mean':(123.675, 116.28, 103.53),
'std':(1.0,1.0,1.0),#(58.395, 57.12, 57.375),
'lr_steps': (80000, 100000,120000),
'max_iter': 120000,
'max_epoch':80,
'feature_maps': [38, 19, 10, 5, 3, 1],
'min_dim': 300,
'backbone_out':[512,1024,2048,512,256,256],
'neck_out':[256,256,256,256,256,256],
'steps':[8, 16, 32, 64, 100, 300],
'min_sizes': [30, 60, 111, 162, 213, 264],
'max_sizes': [60, 111, 162, 213, 264, 315],
'aspect_ratios': [[2], [2, 3], [2, 3], [2, 3], [2], [2]],
'variance': [0.1, 0.2],
'clip': True,
'nms_kind': "cluster_weighted_diounms", #Currently, NMS only surports 'cluster_nms', 'cluster_diounms', 'cluster_weighted_nms', 'cluster_weighted_diounms'
'beta1':0.5,
'name': 'VOC',
'work_name':"SSD300_VOC_FPN_GIOU",
}
crack = {
'model':"resnet50",
'num_classes': 2,
'mean':(127.5, 127.5, 127.5),
'std':(1.0, 1.0, 1.0),
'lr_steps': (25000, 35000, 45000),
'max_iter': 50000,
'max_epoch':2000,
'feature_maps': [38, 19, 10, 5, 3, 1],
'min_dim': 300,
'backbone_out':[512,1024,2048,512,256,256],
'neck_out':[256,256,256,256,256,256],
'steps': [8, 16, 32, 64, 100, 300],
'min_sizes': [21, 45, 99, 153, 207, 261],
'max_sizes': [45, 99, 153, 207, 261, 315],
'aspect_ratios': [[2], [2, 3], [2, 3], [2, 3], [2], [2]],#[[1/0.49], [1/0.16,1/0.09], [1/0.16,1/0.09], [1/0.16,1/0.09], [1/0.09], [1/0.09]],
'variance': [0.1, 0.2],
'clip': True,
'name': 'CRACK',
'work_name':"SSD300_CRACK_FPN_GIOU",
}
coco = {
'num_classes': 201,
'lr_steps': (280000, 360000, 400000),
'max_iter': 400000,
'max_epoch':80,
'feature_maps': [38, 19, 10, 5, 3, 1],
'min_dim': 300,
'steps': [8, 16, 32, 64, 100, 300],
'min_sizes': [21, 45, 99, 153, 207, 261],
'max_sizes': [45, 99, 153, 207, 261, 315],
'aspect_ratios': [[2], [2, 3], [2, 3], [2, 3], [2], [2]],
'variance': [0.1, 0.2],
'clip': True,
'name': 'COCO',
}
trafic = {
'num_classes': 21,
'lr_steps': (80000, 100000, 120000),
'max_iter': 12000,
'max_epoch':2000,
'feature_maps': [50, 25, 13, 7, 5, 3],
'min_dim': 800,
'steps': [16, 32, 64, 100, 300, 600],
'min_sizes': [16, 32, 64, 128, 256, 512],
'max_sizes': [32, 64, 128, 256, 512, 630],
'aspect_ratios': [[1], [1,1/2], [1/2,1], [1/2,1], [1], [1]],
'variance': [0.1, 0.2],
'clip': True,
'name': 'TRAFIC',
}