RevDra commited on
Commit
8d42e8f
·
verified ·
1 Parent(s): 28068d8

Update src/face_detection_yolov12.py

Browse files
Files changed (1) hide show
  1. src/face_detection_yolov12.py +2 -2
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
- # 1. Cấu hình an toàn cho PyTorch (Bắt buộc cho model mới)
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"Loaded YOLOv12 model from {model_path}")
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: