algo

Signed-off-by: Saurabh Misra <[email protected]>

Author: misrasaurabh1

codeflash/after_algo.py

@@ -0,0 +1,9 @@
+def postprocess(self, predictions: tuple[np.ndarray, ...], max_detections: int):
+    bboxes, logits = predictions
+    batch_size, num_queries, num_classes = logits.shape
+    logits_sigmoid = self.sigmoid_stable(logits)
+    for batch_idx in range(batch_size):
+        logits_flat = logits_sigmoid[batch_idx].reshape(-1)
+        # Use argpartition for better performance when max_detections is smaller than logits_flat
+        partition_indices = np.argpartition(-logits_flat, max_detections)[:max_detections]
+        sorted_indices = partition_indices[np.argsort(-logits_flat[partition_indices])]

codeflash/before_algo.py

@@ -0,0 +1,7 @@
+def postprocess(self, predictions: tuple[np.ndarray, ...], max_detections: int):
+    bboxes, logits = predictions
+    batch_size, num_queries, num_classes = logits.shape
+    logits_sigmoid = self.sigmoid_stable(logits)
+    for batch_idx in range(batch_size):
+        logits_flat = logits_sigmoid[batch_idx].reshape(-1)
+        sorted_indices = np.argsort(-logits_flat)[:max_detections]