VideoAnalysis/neuralNetwork.py

10 lines
280 B
Python
Raw Normal View History

2023-06-27 00:28:41 +04:00
from ultralytics import YOLO
def load_model(name: str = 'yolov8s.pt') -> YOLO:
'''
2023-06-27 00:28:41 +04:00
Загрузка предварительно натренированной модели.
@param name: Название модели.
'''
2023-06-27 00:28:41 +04:00
model = YOLO(name);
return model