db config...
{'ae_threshold': 0.3,
'att_max_crops': 30,
'att_nms_ks': [3, 3, 3],
'att_ranges': [[96, 256], [32, 96], [0, 32]], #对应大中小三个尺度的范围
'att_ratios': [16, 8, 4],
'att_scales': [[1, 2, 4]], #对大中小三种目标的缩放级别,对更小的目标缩放更大
'att_sizes': [[16, 16], [32, 32], [64, 64]], #输出的三个att map的尺寸,分别为输入大小除以16,8,4,对应==att_ratios==
'att_thresholds': [0.3],
'border': 64,
'categories': 50,
'data_aug': True,
'gaussian_bump': True,
'gaussian_iou': 0.5,
'gaussian_radius': -1,
'init_sizes': [192, 255], #test阶段寻找,下采样估计目标位置时的缩放尺寸
'input_size': [255, 255], #**输入大小**,裁剪方式输入
'lighting': True,
'max_per_image': 100,
'max_per_set': 40,
'max_scale': 32,
'merge_bbox': False,
'min_scale': 16,
'nms_algorithm': 'exp_soft_nms',
'nms_kernel': 3,
'nms_threshold': 0.5,
'num_dets': 12,
'output_sizes': [[64, 64]], #HourGlass 输出特征图的尺寸,输入大小除以4(即,heat map,embedding, offset三个分支的大小)
'rand_center': True,
'rand_color': False,
'rand_crop': False,
'rand_scale_max': 1.1,
'rand_scale_min': 0.5,
'rand_scale_step': 0.1,
'rand_scales': array([0.5, 0.6, 0.7, 0.8, 0.9, 1. , 1.1]),
'ref_dets': True,
'score_threshold': 0.05,
'test_flipped': True,
'test_scales': [1],
'top_k': 12,
'view_sizes': [],
'weight_exp': 8}
CornerNet-Lite参数说明
最新推荐文章于 2021-11-11 10:19:15 发布