(self, video, pred_tracks, pred_visibility, filename="visual_trace.mp4", mode="ranbow")
| 54 | |
| 55 | |
| 56 | def visualize(self, video, pred_tracks, pred_visibility, filename="visual_trace.mp4", mode="ranbow"): |
| 57 | if mode == "rainbow": |
| 58 | self.visualizer.color_map = cm.get_cmap("gist_rainbow") |
| 59 | elif mode == "cool": |
| 60 | self.visualizer.color_map = cm.get_cmap(mode) |
| 61 | return self.visualizer.visualize( |
| 62 | video, |
| 63 | pred_tracks, |
| 64 | pred_visibility, |
| 65 | query_frame=0 if self.backward_tracking else self.grid_query_frame, |
| 66 | filename=filename, |
| 67 | ) |
| 68 | |
| 69 | @classmethod |
| 70 | def cluster_traces(self, traces, n_clusters=3): |
no outgoing calls
no test coverage detected