Commit 5e33ca5

committed

Adds Vit tracker option

1 parent 0658d11 commit 5e33ca5Copy full SHA for 5e33ca5

File tree

3 files changed

+14

-2

lines changed

README.md
model/vit
- object_tracking_vittrack_2023sep.onnx
performance_comparison_sample.py

3 files changed

+14

-2

lines changed

`‎README.md`

Lines changed: 5 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -7,9 +7,10 @@ Python版OpenCVのTracking APIのサンプルです。<br>`
`7`	`7`	`* opencv-contrib-python 4.8.0.74 or later`
`8`	`8`
`9`	`9`	`# Algorithm`
`10`		`-2023年07月25日時点でOpenCVには以下10アルゴリズムが実装されています。`
	`10`	`+2024年01月18日時点でOpenCVには以下11アルゴリズムが実装されています。`
`11`	`11`	`* DaSiamRPN`
`12`	`12`	`* NanoTrack`
	`13`	`+* Vit`
`13`	`14`	`* MIL`
`14`	`15`	`* GOTURN`
`15`	`16`	`* CSRT`
`@@ -68,6 +69,9 @@ DaSiamRPNトラッカーの使用有無<br>`
`68`	`69`	`* --use_nano<br>`
`69`	`70`	`NanoTrackの使用有無<br>`
`70`	`71`	`デフォルト:指定なし`
	`72`	`+* --use_vit<br>`
	`73`	`+Vitの使用有無<br>`
	`74`	`+デフォルト:指定なし`
`71`	`75`	`* --use_csrt<br>`
`72`	`76`	`CSRTトラッカーの使用有無<br>`
`73`	`77`	`デフォルト:指定なし`

`‎model/vit/object_tracking_vittrack_2023sep.onnx`

698 KB

Binary file not shown.

`‎performance_comparison_sample.py`

Lines changed: 9 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -26,6 +26,7 @@ def get_args():`
`26`	`26`	`parser.add_argument('--use_medianflow', action='store_true')`
`27`	`27`	`parser.add_argument('--use_tld', action='store_true')`
`28`	`28`	`parser.add_argument('--use_nano', action='store_true')`
	`29`	`+ parser.add_argument('--use_vit', action='store_true')`
`29`	`30`
`30`	`31`	`args = parser.parse_args()`
`31`	`32`
`@@ -63,6 +64,10 @@ def initialize_tracker_list(window_name, image, tracker_algorithm_list):`
`63`	`64`	`# params.backbone = "model/nanotrackv3/nanotrack_backbone_sim.onnx"`
`64`	`65`	`# params.neckhead = "model/nanotrackv3/nanotrack_head_sim.onnx"`
`65`	`66`	`tracker = cv.TrackerNano_create(params)`
	`67`	`+ if tracker_algorithm == 'Vit':`
	`68`	`+ params = cv.TrackerVit_Params()`
	`69`	`+ params.net = "model/vit/object_tracking_vittrack_2023sep.onnx"`
	`70`	`+ tracker = cv.TrackerVit_create(params)`
`66`	`71`	`if tracker_algorithm == 'CSRT':`
`67`	`72`	`tracker = cv.TrackerCSRT_create()`
`68`	`73`	`if tracker_algorithm == 'KCF':`
`@@ -125,6 +130,7 @@ def main():`
`125`	`130`	`use_medianflow = args.use_medianflow`
`126`	`131`	`use_tld = args.use_tld`
`127`	`132`	`use_nano = args.use_nano`
	`133`	`+ use_vit = args.use_vit`
`128`	`134`
`129`	`135`	`# 使用アルゴリズム #########################################################`
`130`	`136`	`tracker_algorithm_list = []`
`@@ -148,7 +154,9 @@ def main():`
`148`	`154`	`tracker_algorithm_list.append('TLD')`
`149`	`155`	`if use_nano:`
`150`	`156`	`tracker_algorithm_list.append('Nano')`
`151`		`-`
	`157`	`+ if use_vit:`
	`158`	`+ tracker_algorithm_list.append('Vit')`
	`159`	`+`
`152`	`160`	`if len(tracker_algorithm_list) == 0:`
`153`	`161`	`tracker_algorithm_list.append('DaSiamRPN')`
`154`	`162`	`print(tracker_algorithm_list)`

0 commit comments

Comments

(0)

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit 5e33ca5

File tree

3 files changed

3 files changed

`‎README.md`

`‎model/vit/object_tracking_vittrack_2023sep.onnx`

`‎performance_comparison_sample.py`

0 commit comments