| 341 | |
| 342 | |
| 343 | class TaskInfo: |
| 344 | def __init__(self): |
| 345 | self.frames = None |
| 346 | self.id = -1 |
| 347 | self.bboxes = None |
| 348 | self.action_preds = None |
| 349 | self.num_buffer_frames = 0 |
| 350 | self.img_height = -1 |
| 351 | self.img_width = -1 |
| 352 | self.crop_size = -1 |
| 353 | self.clip_vis_size = -1 |
| 354 | |
| 355 | def add_frames(self, idx, frames): |
| 356 | """ |
| 357 | Add the clip and corresponding id. |
| 358 | Args: |
| 359 | idx (int): the current index of the clip. |
| 360 | frames (list[ndarray]): list of images in "BGR" format. |
| 361 | """ |
| 362 | self.frames = frames |
| 363 | self.id = idx |
| 364 | |
| 365 | def add_bboxes(self, bboxes): |
| 366 | """ |
| 367 | Add correspondding bounding boxes. |
| 368 | """ |
| 369 | self.bboxes = bboxes |
| 370 | |
| 371 | def add_action_preds(self, preds): |
| 372 | """ |
| 373 | Add the corresponding action predictions. |
| 374 | """ |
| 375 | self.action_preds = preds |