Add visibility/mps to notebook demo (#16)
* add changes to notebook, allow None visibilities in visualizer
This commit is contained in:
		| @@ -226,6 +226,9 @@ class Visualizer: | |||||||
|         for t in range(T): |         for t in range(T): | ||||||
|             for i in range(N): |             for i in range(N): | ||||||
|                 coord = (tracks[t, i, 0], tracks[t, i, 1]) |                 coord = (tracks[t, i, 0], tracks[t, i, 1]) | ||||||
|  |                 visibile = True | ||||||
|  |                 if visibility is not None: | ||||||
|  |                     visibile = visibility[0, t, i] | ||||||
|                 if coord[0] != 0 and coord[1] != 0: |                 if coord[0] != 0 and coord[1] != 0: | ||||||
|                     if not compensate_for_camera_motion or ( |                     if not compensate_for_camera_motion or ( | ||||||
|                         compensate_for_camera_motion and segm_mask[i] > 0 |                         compensate_for_camera_motion and segm_mask[i] > 0 | ||||||
| @@ -236,7 +239,7 @@ class Visualizer: | |||||||
|                             coord, |                             coord, | ||||||
|                             int(self.linewidth * 2), |                             int(self.linewidth * 2), | ||||||
|                             vector_colors[t, i].tolist(), |                             vector_colors[t, i].tolist(), | ||||||
|                             thickness=-1 if visibility[0, t, i] else 2 |                             thickness=-1 if visibile else 2 | ||||||
|                             -1, |                             -1, | ||||||
|                         ) |                         ) | ||||||
|  |  | ||||||
|   | |||||||
										
											
												File diff suppressed because one or more lines are too long
											
										
									
								
							
		Reference in New Issue
	
	Block a user