make ruff happy

This commit is contained in:
SkalskiP committed 2024-01-04 15:52:40 +01:00
1 parent bdaee3f1fc
commit be57c8872d
2 files changed
+8 -6

No files matched your search

@@ -68,12 +68,13 @@ def detect(
frame: np.ndarray, model: RoboflowInferenceModel, confidence_threshold: float = 0.5
) -> sv.Detections:
"""
Detect objects in a frame using a Inference model, filtering detections by class ID and
confidence threshold.
Detect objects in a frame using Inference model, filtering detections by class ID
and confidence threshold.
Args:
frame (np.ndarray): The frame to process, expected to be a NumPy array.
model (RoboflowInferenceModel): The Inference model used for processing the frame.
model (RoboflowInferenceModel): The Inference model used for processing the
frame.
confidence_threshold (float, optional): The confidence threshold for filtering
detections. Default is 0.5.
@@ -82,8 +83,8 @@ def detect(
model.
Note:
This function is specifically tailored for an Inference model and assumes class ID 0
for filtering.
This function is specifically tailored for an Inference model and assumes class
ID 0 for filtering.
"""
results = model.infer(frame)[0]
detections = sv.Detections.from_inference(results)
+2 -1
View File
@@ -448,7 +448,8 @@ class Detections:
def from_roboflow(cls, roboflow_result: Union[dict, Any]) -> Detections:
"""
Create a Detections object from the [Roboflow](https://roboflow.com/)
API inference result or the [Inference](https://inference.roboflow.com/) package results.
API inference result or the [Inference](https://inference.roboflow.com/)
package results.
Args:
roboflow_result (dict): The result from the