RizwanMunawar/yolov7-object-tracking

YOLOv7 Object Tracking Using PyTorch, OpenCV and Sort Tracking

解決的問題

本專案提供了一種在影片串流中執行即時目標檢測與追蹤的方法。它不僅允許使用者識別目標是什麼,還能在目標穿過影片的不同影格時,為該目標維持唯一的身份識別。

工作原理

該系統結合了用於定位每一影格中目標的檢測模型(主要是 YOLOv7,並支援 YOLOv8)以及用於為這些目標分配並維持 ID 的追蹤演算法(參考 SORT)。它可以處理來自各種來源的輸入,包括影片檔案、網路攝影機、外部攝影機和 IP 攝影機串流。

適用對象

這適用於需要在其應用程式中實現目標追蹤的開發人員與電腦視覺工程師,例如用於監控、交通監測或任何需要計數或追蹤特定目標的場景。

亮點

  • 多模型支援:支援 YOLOv7 與 YOLOv8 權重。
  • 靈活的輸入來源:支援影片檔案、網路攝影機、IP 串流與外部攝影機。
  • 詳細的輸出:能夠將追蹤質心、目標 ID 與邊界框維度儲存至文字檔中。
  • 可自定義的追蹤:支援按特定目標類別進行過濾(例如,僅追蹤人)以及用於不同追蹤軌跡的獨特顏色等視覺輔助功能。

相關

  • 專案
  • 專案
  • 專案
  • 專案
  • 專案