12 lines
338 B
Python
12 lines
338 B
Python
|
import warnings
|
||
|
|
warnings.filterwarnings('ignore')
|
||
|
|
from ultralytics import YOLO
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
model = YOLO('yolov8n.pt') # select your model.pt path
|
||
|
|
model.track(source='video.mp4',
|
||
|
|
imgsz=640,
|
||
|
|
project='runs/track',
|
||
|
|
name='exp',
|
||
|
|
save=True
|
||
|
|
)
|