VideoAnalysis/neuralNetwork.py
2023-06-27 00:28:41 +04:00

10 lines
280 B
Python

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