| 137 | |
| 138 | |
| 139 | class SoftNMSer(object): |
| 140 | def __init__(self, method, gaussian_sigma, linear_threshold, prune_threshold): |
| 141 | self.method = method |
| 142 | self.gaussian_sigma = gaussian_sigma |
| 143 | self.linear_threshold = linear_threshold |
| 144 | self.prune_threshold = prune_threshold |
| 145 | |
| 146 | def __call__(self, boxes, scores, class_idxs): |
| 147 | return batched_soft_nms(boxes, scores, class_idxs, self.method, self.gaussian_sigma, self.linear_threshold, self.prune_threshold) |
| 148 | |
| 149 | |
| 150 | def batched_soft_nms_rotated( |