(filename, results)
| 51 | |
| 52 | |
| 53 | def write_results_score(filename, results): |
| 54 | save_format = '{frame},{id},{x1},{y1},{w},{h},{s},-1,-1,-1\n' |
| 55 | with open(filename, 'w') as f: |
| 56 | for i in range(results.shape[0]): |
| 57 | frame_data = results[i] |
| 58 | frame_id = int(frame_data[0]) |
| 59 | track_id = int(frame_data[1]) |
| 60 | x1, y1, w, h = frame_data[2:6] |
| 61 | score = frame_data[6] |
| 62 | line = save_format.format(frame=frame_id, id=track_id, x1=x1, y1=y1, w=w, h=h, s=-1) |
| 63 | f.write(line) |
| 64 | |
| 65 | |
| 66 | def dti(txt_path, save_path, n_min=25, n_dti=20): |