def write_output_format_file(_output_format_file, _output_format_dict): _fid = open(_output_format_file, 'w') # _fid.write('{},{},{},{}\n'.format('frame number with 20ms hop (int)', 'class index (int)', 'azimuth angle (int)', 'elevation angle (int)')) for _frame_ind in _output_format_dict.keys(): for _value in _output_format_dict[_frame_ind]: if len(_value) == 4: # Write Cartesian format output. Since baseline does not estimate track count we use a fixed value. _fid.write('{},{},{},{},{},{}\n'.format(int(_frame_ind)+1, int(_value[0]), 0, float(_value[1]), float(_value[2]), float(_value[3]))) else: # Write Polar format output. _fid.write('{},{},{},{},{},{}\n'.format(int(_frame_ind)+1, int(_value[0]), 0, int(_value[1]), int(_value[2]))) _fid.close()