Fix bugs

ozora-ogino · Jan 8, 2022 · aa1b43e · aa1b43e
1 parent 423c00f
commit aa1b43e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/detect.py b/src/detect.py
@@ -68,7 +68,7 @@ def postprocess(self, output_data, box_type: str) -> Tuple[np.ndarray]:
 
         if box_type == "xyxy":
             # xywh -> xyxyx
-            boxes = self.to_xyxy(box_type)
+            boxes = self.to_xyxy(boxes)
 
         # Filter by confidence threshold.
         idxs = np.where(conf > self.conf_thr)