def read_det_file(det_filename): """ Parse lines in 2D detection output files """ det_id2str = {1: "Pedestrian", 2: "Car", 3: "Cyclist"} objects = {} with open(det_filename, "r") as f: for line in f.readlines(): obj = utils.Object2d(line.rstrip()) if obj.img_name not in objects.keys(): objects[obj.img_name] = [] objects[obj.img_name].append(obj) # objects = [utils.Object2d(line.rstrip()) for line in f.readlines()] return objects
def read_det_file(det_filename): ''' Parse lines in 2D detection output files ''' det_id2str = {1: 'Pedestrian', 2: 'Car', 3: 'Cyclist'} objects = {} with open(det_filename, 'r') as f: for line in f.readlines(): obj = utils.Object2d(line.rstrip()) if obj.img_name not in objects.keys(): objects[obj.img_name] = [] objects[obj.img_name].append(obj) #objects = [utils.Object2d(line.rstrip()) for line in f.readlines()] return objects