22 lines
662 B
Python
22 lines
662 B
Python
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()
|