(self, model_path, max_dist=0.1, min_confidence=0.3, nms_max_overlap=1.0, max_iou_distance=0.7, max_age=30, n_init=3, nn_budget=100, use_cuda=True)
| 154 | |
| 155 | class DeepSort(object): |
| 156 | def __init__(self, model_path, max_dist=0.1, min_confidence=0.3, nms_max_overlap=1.0, max_iou_distance=0.7, max_age=30, n_init=3, nn_budget=100, use_cuda=True): |
| 157 | self.min_confidence = min_confidence |
| 158 | self.nms_max_overlap = nms_max_overlap |
| 159 | |
| 160 | self.extractor = Extractor(model_path, use_cuda=use_cuda) |
| 161 | |
| 162 | max_cosine_distance = max_dist |
| 163 | metric = NearestNeighborDistanceMetric( |
| 164 | "cosine", max_cosine_distance, nn_budget) |
| 165 | self.tracker = Tracker( |
| 166 | metric, max_iou_distance=max_iou_distance, max_age=max_age, n_init=n_init) |
| 167 | |
| 168 | def update(self, output_results, img_info, img_size, img_file_name): |
| 169 | img_file_name = os.path.join(get_yolox_datadir(), 'mot', 'train', img_file_name) |
nothing calls this directly
no test coverage detected