siamese-fc pytorch代码运行配置

siamese-fc pytorch代码运行配置

工程文件夹:
siamese-fc pytorch代码运行配置_第1张图片
Car1下面为
siamese-fc pytorch代码运行配置_第2张图片
1.代码下载地址:

https://github.com/StrangerZhang/SiamFC-PyTorch

2.下载模型2016-08-17.net.mat,放入新建的models文件夹下

http://www.robots.ox.ac.uk/%7Eluca/stuff/siam-fc_nets/2016-08-17.net.mat

3.pycharm打开工程,需要配置好pytorch,opencv等库
4.在bin文件夹,运行convert_pretrained_model.py
转换mat权重文件为siamfc_pretrained.pth
5.修改demo_siamfc.py最下方代码:

if __name__ == "__main__":
    video_dir = '../Car1/'
    gpu_id = 0
    model_path = '../models/siamfc_pretrained.pth'
    main(video_dir, gpu_id, model_path)

6.运行demo_siamfc.py
7.效果如下:

siamfc目标跟踪视频

car1视频下载地址:
https://pan.baidu.com/s/19aD7htuGUO7ERr7-zmTBmQ
提取码:n3kn

你可能感兴趣的:(目标跟踪)