siamese-fc pytorch代码运行配置
工程文件夹:
Car1下面为
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目标跟踪视频