renamed scripts folder to pykick
This commit is contained in:
21
pykick/detection_demo.py
Normal file
21
pykick/detection_demo.py
Normal file
@@ -0,0 +1,21 @@
|
||||
from __future__ import print_function
|
||||
from __future__ import division
|
||||
|
||||
from .utils import read_config
|
||||
from .imagereaders import NaoImageReader, VideoReader
|
||||
from .finders import BallFinder
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
video = VideoReader(0, loop=True)
|
||||
hsv = read_config()
|
||||
hsv_lower = tuple(map(hsv.get, ('low_h', 'low_s', 'low_v')))
|
||||
hsv_upper = tuple(map(hsv.get, ('high_h', 'high_s', 'high_v')))
|
||||
finder = BallFinder(hsv_lower, hsv_upper, hsv['min_radius'], None)
|
||||
try:
|
||||
while True:
|
||||
frame = video.get_frame()
|
||||
finder.find_colored_ball(frame)
|
||||
finder.visualize(frame)
|
||||
finally:
|
||||
video.close()
|
||||
Reference in New Issue
Block a user