Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- 1.
- !pip install ultralytics
- 2.
- from ultralytics import YOLO
- 3. #Трекинг и детекция
- # подставить своё видео
- model = YOLO('yolov8n.pt')
- results = model.track(source="https://www.youtube.com/watch?v=NlPt4YkbLyk", save=True, stream_buffer=True)
- 4. #Делаем красиво. Сначала конвертируем полученное видео. Нужно у прошлого результата вставить путь сюда
- # например путь /content/runs/detect/predict/watch_v_NlPt4YkbLyk.avi
- # в итоге будет
- !ffmpeg -i /content/runs/detect/predict/watch_v_NlPt4YkbLyk.avi output.mp4
- 5.
- from IPython.display import HTML
- from base64 import b64encode
- mp4 = open('output.mp4','rb').read()
- data_url = "data:video/mp4;base64," + b64encode(mp4).decode()
- 6.
- HTML("""
- <video controls>
- <source src="%s" type="video/mp4">
- </video>
- """ % data_url)
- 7. #Трекинг и позиция
- # подставить своё видео
- model = YOLO('yolov8n-pose.pt')
- results = model.track(source="https://www.youtube.com/watch?v=NlPt4YkbLyk", save=True, stream_buffer=True)
- 8. #Красивости делаем по той же схеме
Add Comment
Please, Sign In to add comment