Exemplo n.º 1
0
 def _viz_tracklets(self):
     print("viz tracklets frame", self._time_idx)
     import matplotlib.pyplot as plt
     from matplotlib.patches import Rectangle
     if self._ax is None:
         fig, self._ax = plt.subplots(1)
     colors = generate_colors()
     t = self._time_idx
     img = self._imgs_for_viz[-1]
     self._ax.clear()
     self._ax.imshow(img[..., ::-1])
     for idx, tracklet in enumerate(self._all_tracklets):
         # probably filter by confidence and tracklet length
         if tracklet.end_time - tracklet.start_time < 2:
             continue
         if max(tracklet.ff_gt_scores) < 0.2:
             continue
         if tracklet.start_time <= t < tracklet.end_time:
             color = colors[idx % len(colors)]
             box = tracklet.boxes[t - tracklet.start_time]
             width = box[2] - box[0]
             height = box[3] - box[1]
             rect = Rectangle((box[0], box[1]),
                              width,
                              height,
                              color=color,
                              fill=False)
             self._ax.add_patch(rect)
Exemplo n.º 2
0
 def _viz_tracklets(self):
     print("viz tracklets frame", self._time_idx)
     import cv2
     self._cv_img = self._imgs_for_viz[-1].copy()
     colors = generate_colors()
     t = self._time_idx
     for idx, tracklet in enumerate(self._all_tracklets):
         # probably filter by confidence and tracklet length
         #if tracklet.end_time - tracklet.start_time < 2:
         #    continue
         if max(tracklet.ff_gt_scores) < 0.2:
             continue
         if tracklet.start_time <= t < tracklet.end_time:
             color = colors[idx % len(colors)]
             box = tracklet.boxes[t - tracklet.start_time]