diff --git a/supervision/detection/core.py b/supervision/detection/core.py index c5633ea2..1a5c86ea 100644 --- a/supervision/detection/core.py +++ b/supervision/detection/core.py @@ -540,7 +540,7 @@ class Detections: return cls( xyxy=detectron2_results["instances"].pred_boxes.tensor.cpu().numpy(), confidence=detectron2_results["instances"].scores.cpu().numpy(), - mask=detectron2_results['instances'].pred_masks.detach().cpu().numpy(), + mask=detectron2_results["instances"].pred_masks.detach().cpu().numpy(), class_id=detectron2_results["instances"] .pred_classes.cpu() .numpy()