def _get_msg_group(self): msg_group = [] dataset_list = ['eth', 'eth', 'ucy', 'ucy', 'ucy'] dataset_idx_list = [0, 1, 0, 1, 2] #dataset_list = ['eth'] #dataset_idx_list = [0] for i in range(len(dataset_list)): dataset = dataset_list[i] dataset_idx = dataset_idx_list[i] msg = Message() data = DataLoader(dataset, dataset_idx, self.fps) msg = data.update_message(msg) gp = Grouping(msg, self.history) msg = gp.update_message(msg) msg_group.append(msg) return msg_group
group_idx = 20 # Initialize a message msg = Message() # Initialize dataloader data = DataLoader(dataset, dataset_idx) # Update the message msg = data.update_message(msg) # Initialize grouping gp = Grouping(msg, 16) # Update the message msg = gp.update_message(msg) # This shows what group ids are in this frame print(msg.video_labels_matrix[frame_idx]) # Initialize group shape generation gs_gen = GroupShapeGeneration(msg) vertices, pedidx = gs_gen.generate_group_shape(frame_idx, group_idx) # The returned vertices for the group shape print(vertices) print(pedidx) # We can also draw it on an image (blank canvas in this case) canvas = np.zeros((msg.frame_height, msg.frame_width, 3), dtype=np.uint8) dgs = DrawGroupShape(msg) canvas = dgs.draw_group_shape(vertices, canvas) cv2.imwrite('example.jpg', canvas)