Extended colorpicker to read pictures also

Some minor refactoring on photo capturing script, also.
This commit is contained in:
2018-06-01 20:11:56 +02:00
parent ec90e8fda8
commit cc8589d750
3 changed files with 39 additions and 45 deletions

View File

@@ -3,7 +3,7 @@ from __future__ import print_function
import json
import argparse
import cv2
from imagereaders import VideoReader, NaoImageReader
from imagereaders import VideoReader, NaoImageReader, PictureReader
# import imutils
class Colorpicker(object):
@@ -89,6 +89,10 @@ if __name__ == '__main__':
'--video-file',
help='video file to use'
)
parser.add_argument(
'--image-file',
help='image to use'
)
parser.add_argument(
'--still',
help='only take one image from video stream',
@@ -110,6 +114,8 @@ if __name__ == '__main__':
cp.load(args.input_config)
if args.video_file:
rdr = VideoReader(args.video_file, loop=True)
elif args.image_file:
rdr = PictureReader(args.image_file)
elif args.nao_ip:
rdr = NaoImageReader(
args.nao_ip,