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

Goal detection bugs fixed

Browse Source
This commit is contained in:
Pavel Lutskov
2018-06-23 14:11:02 +02:00
parent abb9750413
commit 700c7a73c7
4 changed files with 22 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pykick/detection_demo.py
View File

@@ -5,7 +5,7 @@ import argparse
import cv2
from .utils import read_config, imresize, field_mask
from .utils import read_config, imresize
from .imagereaders import NaoImageReader, VideoReader, PictureReader
from .finders import BallFinder, GoalFinder, FieldFinder
@@ -100,10 +100,9 @@ if __name__ == '__main__':
frame = imresize(frame, width=args.width)
field = field_finder.find(frame)
not_field = cv2.bitwise_not(field)
ball_frame = field_finder.draw(frame, field)
goal_frame = field_finder.draw(frame, not_field)
ball_frame = field_finder.mask_it(frame, field)
goal_frame = field_finder.mask_it(frame, field, inverse=True)
ball = ball_finder.find(ball_frame)
goal = goal_finder.find(goal_frame)