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)
