siamfc运行_sta系统

大家好，又见面了，我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全系列IDE稳定放心使用

修改siamfc文件夹下的Load.py文件

import sys
sys.path.append('/home/nanorobot/Documents/siamfc/siamfc')
sys.path.append('/home/nanorobot/Documents/siamfc')
from siamfc import TrackerSiamFC, ops

上面是把需要用到的自定义的模块路径引入进来，以防后面找不到模块，路径根据你自己的文件位置改。

下面是修改主函数：

if __name__ == '__main__':
    seq_dir = os.path.expanduser('/home/nanorobot/Documents/siamfc/BlurBody/')
    img_files = sorted(glob.glob(seq_dir + 'datasets/*.jpg'))
    anno = np.loadtxt(seq_dir + 'groundtruth.txt')

    ims = [cv2.imread(imf) for imf in img_files]

    VeryBig = 999999999
    Cap = cv2.VideoCapture(0)
    ret,frame = Cap.read()
    ims = [frame]
    cv2.namedWindow('SiamFC', cv2.WND_PROP_FULLSCREEN)
    init_rect = cv2.selectROI("SiamFC", ims[0], False, False)
    cv2.waitKey(0)
    x, y, w, h = init_rect
    print(x, y, w, h)

    toc = 0
    im = frame
    target_pos = np.array([x+w/2, y+h/2])
    target_sz = np.array([w, h])
    net_path = seq_dir + 'siamfc_alexnet_e50.pth'
    tracker = TrackerSiamFC(net_path=net_path)
    tracker.init(frame, init_rect)
    while(True):
        ret, im = Cap.read()
        if (ret==False):
            break
        box = tracker.update(im)
        ops.show_image(im, box,cvt_code=None)

修改好后，运行

python Load.py

显示
选定目标

按下回车或者空格，等待一会儿就可以了