sobakasutulaya

Трекинг детекция и трекинг позиция

Nov 7th, 2023
11
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.21 KB | None | 0 0
  1. 1.
  2. !pip install ultralytics
  3.  
  4. 2.
  5. from ultralytics import YOLO
  6.  
  7. 3. #Трекинг и детекция
  8. # подставить своё видео
  9. model = YOLO('yolov8n.pt')
  10. results = model.track(source="https://www.youtube.com/watch?v=NlPt4YkbLyk", save=True, stream_buffer=True)
  11.  
  12.  
  13. 4. #Делаем красиво. Сначала конвертируем полученное видео. Нужно у прошлого результата вставить путь сюда
  14. # например путь /content/runs/detect/predict/watch_v_NlPt4YkbLyk.avi
  15. # в итоге будет
  16. !ffmpeg -i /content/runs/detect/predict/watch_v_NlPt4YkbLyk.avi output.mp4
  17.  
  18. 5.
  19. from IPython.display import HTML
  20. from base64 import b64encode
  21. mp4 = open('output.mp4','rb').read()
  22. data_url = "data:video/mp4;base64," + b64encode(mp4).decode()
  23.  
  24. 6.
  25. HTML("""
  26. <video controls>
  27. <source src="%s" type="video/mp4">
  28. </video>
  29. """ % data_url)
  30.  
  31. 7. #Трекинг и позиция
  32. # подставить своё видео
  33. model = YOLO('yolov8n-pose.pt')
  34. results = model.track(source="https://www.youtube.com/watch?v=NlPt4YkbLyk", save=True, stream_buffer=True)
  35.  
  36. 8. #Красивости делаем по той же схеме
Add Comment
Please, Sign In to add comment