| | |
| | import numpy as np |
| | from typing import List |
| |
|
| | from detectron2.structures import Instances |
| |
|
| |
|
| | def create_prediction_pairs( |
| | instances: Instances, |
| | prev_instances: Instances, |
| | iou_all: np.ndarray, |
| | threshold: float = 0.5, |
| | ) -> List: |
| | """ |
| | Args: |
| | instances: predictions from current frame |
| | prev_instances: predictions from previous frame |
| | iou_all: 2D numpy array containing iou for each bbox pair |
| | threshold: below the threshold, doesn't consider the pair of bbox is valid |
| | Return: |
| | List of bbox pairs |
| | """ |
| | bbox_pairs = [] |
| | for i in range(len(instances)): |
| | for j in range(len(prev_instances)): |
| | if iou_all[i, j] < threshold: |
| | continue |
| | bbox_pairs.append( |
| | { |
| | "idx": i, |
| | "prev_idx": j, |
| | "prev_id": prev_instances.ID[j], |
| | "IoU": iou_all[i, j], |
| | "prev_period": prev_instances.ID_period[j], |
| | } |
| | ) |
| | return bbox_pairs |
| |
|
| |
|
| | LARGE_COST_VALUE = 100000 |
| |
|
