Skip to content

Latest commit

 

History

History

siam-mot

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

SiamMOT

Input

Input

(Video from https://vimeo.com/60139361)

Output

Output

Usage

Automatically downloads the onnx and prototxt files on the first run. It is necessary to be connected to the Internet while downloading.

For the sample video,

$ python3 siam-mot.py

If you want to specify the video image, put the file path after the --video option.
You can use --savepath option to specify the name of the output file to save.

$ python3 siam-mot.py --video VIDEO_PATH --savepath SAVE_FILE_PATH

You can choose person or person_vehicle for track-class such that person tracking or person/vehicle tracking model is used accordingly.

$ python3 siam-mot.py --track-class person

Reference

Framework

Pytorch

Model Format

ONNX opset=11

Netron

person_rpn.onnx.prototxt
person_box.onnx.prototxt
person_track.onnx.prototxt
person_feat_ext.onnx.prototxt
person_vehicle_rpn.onnx.prototxt
person_vehicle_box.onnx.prototxt
person_vehicle_track.onnx.prototxt
person_vehicle_feat_ext.onnx.prototxt