if visualization:
im_show = cv2.cvtColor(im, cv2.COLOR_RGB2BGR)
cv2.rectangle(im_show, (int(target_pos[0] - target_sz[0] / 2),
int(target_pos[1] - target_sz[1] / 2)),
(int(target_pos[0] + target_sz[0] / 2),
int(target_pos[1] + target_sz[1] / 2)),
(0, 255, 0), 3)
cv2.putText(im_show, str(f), (40, 40),
cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 255), 2,
cv2.LINE_AA)
cv2.imshow(video, im_show)
cv2.waitKey(1)
toc = time.time() - tic
fps = n_images / toc
speed.append(fps)
print('{:3d} Video: {:12s} Time: {:3.1f}s\tSpeed: {:3.1f}fps'.format(
video_id, video, toc, fps))
# save result
test_path = join('result', dataset, 'DCFNet_test')
if not isdir(test_path): makedirs(test_path)
result_path = join(test_path, video + '.txt')
with open(result_path, 'w') as f:
for x in res:
f.write(','.join(['{:.2f}'.format(i) for i in x]) + '\n')
print('***Total Mean Speed: {:3.1f} (FPS)***'.format(np.mean(speed)))
eval_auc(dataset, 'DCFNet_test', 0, 1)
if visualization:
im_show = cv2.cvtColor(cv2.imread(frame_name_list[f]),
cv2.COLOR_RGB2BGR)
cv2.rectangle(
im_show,
(int(pos_x - target_w / 2), int(pos_y - target_h / 2)),
(int(pos_x + target_w / 2), int(pos_y + target_h / 2)),
(0, 255, 0), 3)
cv2.putText(im_show, str(f), (40, 40),
cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 255), 2,
cv2.LINE_AA)
cv2.imshow(video, im_show)
cv2.waitKey(1)
toc = time.time() - tic
fps = num_frames / toc
speed.append(fps)
print('{:3d} Video: {:12s} Time: {:3.1f}s\tSpeed: {:3.1f}fps'.format(
video_id, video, toc, fps))
# save result
test_path = os.path.join('result', dataset, 'Siamese-fc_test')
if not os.path.isdir(test_path): os.makedirs(test_path)
result_path = os.path.join(test_path, video + '.txt')
with open(result_path, 'w') as f:
for x in bboxes:
f.write(','.join(['{:.2f}'.format(i) for i in x]) + '\n')
eval_auc(dataset, 'Siamese-fc_test', 0, 1)