yolox-bytetrack-sample

KO | EN

gitlite — search

#react #android #machine-learning #deep-learning #linux #macos #nodejs #java #awesome #awesome-list #vue #pytorch #openai #open-source

yolox-bytetrack-sample

(★ 23)

YOLOXとByteTrackを用いたMOT(Multiple Object Tracking)のPythonサンプル

yolox-bytetrack-sample 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# yolox-bytetrack-sample

[YOLOX](https://github.com/Megvii-BaseDetection/YOLOX)と[ByteTrack](https://github.com/ifzhang/ByteTrack)を用いたMOT(Multiple Object Tracking)のPythonサンプルです。 
YOLOXは[YOLOX-ONNX-TFLite-Sample](https://github.com/Kazuhito00/YOLOX-ONNX-TFLite-Sample)で、ONNXに変換したモデルを使用しています。 
マルチクラス拡張版は[Kazuhito00/yolox-bytetrack-mcmot-sample](https://github.com/Kazuhito00/yolox-bytetrack-mcmot-sample)を参照ください。

https://user-images.githubusercontent.com/37477845/153743142-5715995c-9642-4285-93ce-331f43ec18ef.mp4

# Requirement 
* OpenCV 3.4.2 or later
* onnxruntime 1.5.2 or later
* lap 0.4.0 or later 
* Cython 0.29.27 or later 
* cython_bbox 0.1.3 or later
 
※onnxruntime-gpuはonnxruntimeでも動作しますが、推論時間がかかるためGPUを推奨します 
※Windowsでcython_bbox のインストールが失敗する場合は、GitHubからのインストールをお試しください(2022/02/13時点)

`pip install -e git+https://github.com/samson-wang/cython_bbox.git#egg=cython-bbox`

# Demo
デモの実行方法は以下です。
```bash
python sample.py
```
* --device 
カメラデバイス番号の指定 
デフォルト：0
* --movie 
動画ファイルの指定 ※指定時はカメラデバイスより優先 
デフォルト：指定なし
* --width 
カメラキャプチャ時の横幅 
デフォルト：960
* --height 
カメラキャプチャ時の縦幅 
デフォルト：540
<details>
<summary>YOLOXパラメータ</summary>
 
* --yolox_model 
ロードするモデルの格納パス 
デフォルト：model/yolox_nano.onnx
* --input_shape 
モデルの入力サイズ 
デフォルト：416,416
* --score_th 
クラス判別の閾値 
デフォルト：0.3
* --nms_th 
NMSの閾値 
デフォルト：0.45
* --nms_score_th 
NMSのスコア閾値 
デフォルト：0.1
* --with_p6 
Large P6モデルを使用するか否か 
デフォルト：指定なし
</details>

<details>
<summary>ByteTrackパラメータ</summary>
 
* --track_thresh 
デフォルト：0.5
* --track_buffer 
デフォルト：30
* --match_thresh 
デフォルト：0.8
* --min_box_area 
デフォルト：10
* --mot20 
デフォルト：指定なし
 
※パラメータ詳細は[ByteTrack](https://github.com/ifzhang/ByteTrack)を参照ください。
</details>

# Reference
* [Megvii-BaseDetection/YOLOX](https://github.com/Megvii-BaseDetection/YOLOX)
* [YOLOX-ONNX-TFLite-Sample](https://github.com/Kazuhito00/YOLOX-ONNX-TFLite-Sample)
* [ByteTrack](https://github.com/ifzhang/ByteTrack)

# Author
高橋かずひと(https://twitter.com/KzhtTkhs)
 
# License 
yolox-bytetrack-sample is under [MIT License](LICENSE).

# License(Movie)
サンプル動画は[NHKクリエイティブ・ライブラリー](https://www.nhk.or.jp/archives/creative/)の[ケニア共和国キツイ 町並み(4) ふかんショット](https://www2.nhk.or.jp/archives/creative/material/view.cgi?m=D0002040395_00000)を使用しています。

yolox-bytetrack-sample

yolox-bytetrack-sample 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

yolox-bytetrack-sample

yolox-bytetrack-sample 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README