# bdd_labels = { # 'unlabeled':0, 'dynamic': 1, 'ego vehicle': 2, 'ground': 3, # 'static': 4, 'parking': 5, 'rail track': 6, 'road': 7, # 'sidewalk': 8, 'bridge': 9, 'building': 10, 'fence': 11, # 'garage': 12, 'guard rail': 13, 'tunnel': 14, 'wall': 15, # 'banner': 16, 'billboard': 17, 'lane divider': 18,'parking sign': 19, # 'pole': 20, 'polegroup': 21, 'street light': 22, 'traffic cone': 23, # 'traffic device': 24, 'traffic light': 25, 'traffic sign': 26, 'traffic sign frame': 27, # 'terrain': 28, 'vegetation': 29, 'sky': 30, 'person': 31, # 'rider': 32, 'bicycle': 33, 'bus': 34, 'car': 35, # 'caravan': 36, 'motorcycle': 37, 'trailer': 38, 'train': 39, # 'truck': 40 # } id_dict = {'person': 0, 'rider': 1, 'car': 2, 'bus': 3, 'truck': 4, 'bike': 5, 'motor': 6, 'tl_green': 7, 'tl_red': 8, 'tl_yellow': 9, 'tl_none': 10, 'traffic sign': 11, 'train': 12} id_dict_single = {'car': 0, 'bus': 1, 'truck': 2,'train': 3} # id_dict = {'car': 0, 'bus': 1, 'truck': 2} def convert(size, box): dw = 1./(size[0]) dh = 1./(size[1]) x = (box[0] + box[1])/2.0 y = (box[2] + box[3])/2.0 w = box[1] - box[0] h = box[3] - box[2] x = x*dw w = w*dw y = y*dh h = h*dh return (x,y,w,h)