Spaces:
Runtime error
Runtime error
File size: 5,457 Bytes
6163604 |
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 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 |
# config.py
import os.path
# gets home dir cross platform
HOME = os.path.expanduser("~")
# 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))
MEANS = (246, 246, 246)
exp_cfg = {
'gtdb': {
'num_classes': 2,
'lr_steps': (80000, 100000, 120000),
'max_iter': 120000,
'feature_maps': [64, 32, 16, 8, 4, 2, 1],
'min_dim': 512,
'steps': [8, 16, 32, 64, 128, 256, 512],
'min_sizes': [8.00, 76.8, 153.6, 230.4, 307.2, 384.0, 460.8],
'max_sizes': [76.8, 153.6, 230.4, 307.2, 384.0, 460.8, 537.6],
'aspect_ratios': [[2, 3, 5], [2, 3, 5, 7], [2, 3, 5, 7], [2, 3], [2, 3], [2], [2]],
'variance': [0.1, 0.2],
'clip': True,
'name': 'GTDB',
'is_vertical_prior_boxes_enabled': True,
'mbox': {
'512': [8, 10, 10, 6, 6, 4, 4],
#'512': [5, 6, 6, 4, 4, 3, 3],
'300': [8, 10, 10, 6, 4, 4], # number of boxes per feature map location
},
'extras': {
'512': [256, 'S', 512, 128, 'S', 256, 128, 'S', 256, 128, 'S', 256],
'300': [256, 'S', 512, 128, 'S', 256, 128, 256, 128, 256],
}
},
'math_gtdb_512': {
'num_classes': 2,
'lr_steps': (80000, 100000, 120000),
'max_iter': 240000,
'feature_maps': [64, 32, 16, 8, 4, 2, 1],
'min_dim': 512,
'steps': [8, 16, 32, 64, 128, 256, 512],
'min_sizes': [8.00, 76.8, 153.6, 230.4, 307.2, 384.0, 460.8],
'max_sizes': [76.8, 153.6, 230.4, 307.2, 384.0, 460.8, 537.6],
'aspect_ratios': [[2, 3, 5, 7, 10], [2, 3, 5, 7, 10], [2, 3, 5, 7, 10], [2, 3, 5, 7, 10],
[2, 3, 5, 7, 10], [2, 3, 5, 7, 10], [2, 3, 5, 7, 10]],
'variance': [0.1, 0.2],
'clip': True,
'name': 'math_gtdb_512',
'is_vertical_prior_boxes_enabled': True,
'mbox': {
'512': [12,12,12,12,12,12,12],
},
'extras': {
'512': [256, 'S', 512, 128, 'S', 256, 128, 'S', 256, 128, 'S', 256],
}
},
'ssd300': {
'num_classes': 2,
'lr_steps': (80000, 100000, 120000),
'max_iter': 132000,
'feature_maps': [38, 19, 10, 5, 3, 1],
'min_dim': 300,
'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,
'name': 'ssd300',
'is_vertical_prior_boxes_enabled': True,
'mbox': {
'300': [4, 6, 6, 6, 4, 4], # number of boxes per feature map location
},
'extras': {
'300': [256, 'S', 512, 128, 'S', 256, 128, 256, 128, 256],
}
},
'ssd512': {
'num_classes': 2,
'lr_steps': (80000, 100000, 120000),
'max_iter': 132000,
'feature_maps': [64, 32, 16, 8, 4, 2, 1],
'min_dim': 512,
'steps': [8, 16, 32, 64, 128, 256, 512],
'min_sizes': [35.84, 76.8, 153.6, 230.4, 307.2, 384.0, 460.8],
'max_sizes': [76.8, 153.6, 230.4, 307.2, 384.0, 460.8, 537.6],
'aspect_ratios': [[2], [2, 3], [2, 3], [2, 3], [2,3], [2], [2]],
'variance': [0.1, 0.2],
'clip': True,
'name': 'ssd512',
'is_vertical_prior_boxes_enabled': True,
'mbox': {
'512': [4,6,6,6,6,4,4],
},
'extras': {
'512': [256, 'S', 512, 128, 'S', 256, 128, 'S', 256, 128, 'S', 256],
}
},
'aspect512': {
'num_classes': 2,
'lr_steps': (80000, 100000, 120000),
'max_iter': 132000,
'feature_maps': [64, 32, 16, 8, 4, 2, 1],
'min_dim': 512,
'steps': [8, 16, 32, 64, 128, 256, 512],
'min_sizes': [35.84, 76.8, 153.6, 230.4, 307.2, 384.0, 460.8],
'max_sizes': [76.8, 153.6, 230.4, 307.2, 384.0, 460.8, 537.6],
'aspect_ratios': [[2, 3, 5, 7, 10], [2, 3, 5, 7, 10], [2, 3, 5, 7, 10], [2, 3, 5, 7, 10],
[2, 3, 5, 7, 10], [2, 3, 5, 7, 10], [2, 3, 5, 7, 10]],
'variance': [0.1, 0.2],
'clip': True,
'name': 'ssd512',
'is_vertical_prior_boxes_enabled': True,
'mbox': {
'512': [12,12,12,12,12,12,12],
},
'extras': {
'512': [256, 'S', 512, 128, 'S', 256, 128, 'S', 256, 128, 'S', 256],
}
},
'hboxes512': {
'num_classes': 2,
'lr_steps': (80000, 100000, 120000),
'max_iter': 132000,
'feature_maps': [64, 32, 16, 8, 4, 2, 1],
'min_dim': 512,
'steps': [8, 16, 32, 64, 128, 256, 512],
'min_sizes': [35.84, 76.8, 153.6, 230.4, 307.2, 384.0, 460.8],
'max_sizes': [76.8, 153.6, 230.4, 307.2, 384.0, 460.8, 537.6],
'aspect_ratios': [[2, 3, 5, 7, 10], [2, 3, 5, 7, 10], [2, 3, 5, 7, 10], [2, 3, 5, 7, 10],
[2, 3, 5, 7, 10], [2, 3, 5, 7, 10], [2, 3, 5, 7, 10]],
'variance': [0.1, 0.2],
'clip': True,
'name': 'ssd512',
'is_vertical_prior_boxes_enabled': False,
'mbox': {
'512': [7,7,7,7,7,7,7],
},
'extras': {
'512': [256, 'S', 512, 128, 'S', 256, 128, 'S', 256, 128, 'S', 256],
}
},
} |