Spaces:
Sleeping
Sleeping
Update src/face_detection_yolov12.py
Browse files
src/face_detection_yolov12.py
CHANGED
|
@@ -5,7 +5,7 @@ Optimized for high FPS performance using YOLOv12 Attention Mechanism
|
|
| 5 |
"""
|
| 6 |
|
| 7 |
import torch
|
| 8 |
-
|
| 9 |
try:
|
| 10 |
from ultralytics.nn.tasks import DetectionModel
|
| 11 |
torch.serialization.add_safe_globals([DetectionModel])
|
|
@@ -27,7 +27,7 @@ class YOLOv12FaceDetector:
|
|
| 27 |
try:
|
| 28 |
from ultralytics import YOLO
|
| 29 |
self.yolo = YOLO(model_path)
|
| 30 |
-
print(f"
|
| 31 |
except ImportError:
|
| 32 |
raise ImportError("ultralytics package not found. pip install ultralytics")
|
| 33 |
except Exception as e:
|
|
|
|
| 5 |
"""
|
| 6 |
|
| 7 |
import torch
|
| 8 |
+
|
| 9 |
try:
|
| 10 |
from ultralytics.nn.tasks import DetectionModel
|
| 11 |
torch.serialization.add_safe_globals([DetectionModel])
|
|
|
|
| 27 |
try:
|
| 28 |
from ultralytics import YOLO
|
| 29 |
self.yolo = YOLO(model_path)
|
| 30 |
+
print(f"Loaded YOLOv12 model from {model_path}")
|
| 31 |
except ImportError:
|
| 32 |
raise ImportError("ultralytics package not found. pip install ultralytics")
|
| 33 |
except Exception as e:
|