0%

mac下使用商汤pysot

发表于 更新于 分类于 阅读次数：
本文字数： 1.5k 阅读时长 ≈ 1 分钟

2019年6月4日 下午3:35

三次成功挑战目标跟踪算法极限，商汤开源SiamRPN系列算法

GitHub - STVIR/pysot: SenseTime Research platform for single object tracking, implementing algorithms like SiamRPN and SiamMask.

安装环境：

1
2
3
4
5
6
pip install torch pyyaml yacs tqdm colorama matplotlib cython tensorboardX

python tools/demo.py \
    --config experiments/siamrpn_r50_l234_dwxcorr/config.yaml \
    --snapshot experiments/siamrpn_r50_l234_dwxcorr/model.pth \
    --video demo/bag.avi

注意：请教 · Issue #16 · STVIR/pysot · GitHub
python setup.py build_ext --inplace

每次必执行：

否则：ModuleNotFoundError: No module named ‘pysot’
export PYTHONPATH=/Users/czh/tracker/pysot:$PYTHONPATH

siamrpn_r50_l234_dwxcorr:35fps

1
2
3
4
python tools/demo.py \
    --config experiments/siamrpn_r50_l234_dwxcorr/config.yaml \
    --snapshot experiments/siamrpn_r50_l234_dwxcorr/model.pth \
    --video demo/dataset_rectify_lib_ball/left

siamrpn_alex_dwxcorr:180pfs

1
2
3
4
python tools/demo.py \
    --config experiments/siamrpn_alex_dwxcorr/config.yaml \
    --snapshot experiments/siamrpn_alex_dwxcorr/model.pth \
    --video demo/dataset_rectify_lib_ball/left

siammask_r50_l3:56pfs

1
2
3
4
python tools/demo.py \
    --config experiments/siammask_r50_l3/config.yaml \
    --snapshot experiments/siammask_r50_l3/model.pth \
    --video ../data/dataset_rectify_lib_ball/left

siamrpn_mobilev2_l234_dwxcorr:75fps

1
2
3
4
python tools/demo.py \
    --config experiments/siamrpn_mobilev2_l234_dwxcorr/config.yaml \
    --snapshot experiments/siamrpn_mobilev2_l234_dwxcorr/model.pth \
    --video demo/dataset_rectify_lib_ball/left