(self, image_rgb)
| 218 | return self.detect_model |
| 219 | |
| 220 | def get_face_bboxes(self, image_rgb): |
| 221 | detect_model = self.get_detect_model() |
| 222 | pred = detect_model(image_rgb, conf=0.7, device="") |
| 223 | return pred[0].boxes.xyxy.cpu().numpy() |
| 224 | |
| 225 | def detect_face(self, image_rgb, crop_factor, sort = True): |
| 226 | bboxes = self.get_face_bboxes(image_rgb) |
no test coverage detected