pavel/kick-it
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

modified scripts for presentation

Browse Source
This commit is contained in:
Pavel Lutskov
2018-07-05 10:42:05 +02:00
parent f89864c0e4
commit 4b926bed19
8 changed files with 60 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pykick/detection_demo.py
View File

@@ -4,6 +4,7 @@ from __future__ import division
import argparse
import cv2
import numpy as np
from .utils import read_config, imresize
from .imagereaders import NaoImageReader, VideoReader, PictureReader
@@ -118,9 +119,10 @@ if __name__ == '__main__':
ball_frame = ball_finder.draw(ball_frame, ball)
goal_frame = goal_finder.draw(goal_frame, goal)
combined = np.concatenate((ball_frame, goal_frame), axis=1)
cv2.imshow(ball_window, ball_frame)
cv2.imshow(goal_window, goal_frame)
cv2.imshow(ball_window, combined)
# cv2.imshow(goal_window, goal_frame)
key = cv2.waitKey(0 if args.manual else 1)
if key == ord('q') or key == 27: