大家好,又见面了,我是你们的朋友风君子。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新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
按下回车或者空格,等待一会儿就可以了
最新评论